backward(const Tensor< TInput, MR > &input, const Tensor< TOutput, MR > &output_grad, Tensor< TInput, MR > &input_grad) | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inline |
bias_ | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | private |
config_ | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | private |
createContext(const std::string &device_name) | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprivatestatic |
createOperation() | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inlineprivate |
device_context_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | private |
forward(const Tensor< TInput, MR > &input, Tensor< TOutput, MR > &output) | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inline |
getBias() | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inline |
getDeviceContext() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
getDeviceType() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
getName() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
getParameterTensors() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
getPrecision() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
getStateTensors() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
getWeight() | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inline |
hasBias() const | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inline |
initializeParameterGradients() | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inlineprivate |
initializeParameters() | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inlineprivate |
isTraining() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
Linear(const std::string &device_name, const LinearConfig &config) | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inlineexplicit |
Linear(std::shared_ptr< DeviceContext > device_context, const LinearConfig &config) | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inlineexplicit |
load(ModelArchive &archive) override | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inlinevirtual |
Module(const std::string &device_name, const ComponentConfig &config) | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineexplicit |
Module(std::shared_ptr< DeviceContext > context, const ComponentConfig &config) | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineexplicit |
ModuleBase typedef | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | |
MR typedef | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | |
operation_ | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | private |
output_state_ | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | private |
parameter_grads_ | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | private |
parameter_map_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | protected |
parameterCount() const override | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inlinevirtual |
parameters_ | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | private |
parametersToString() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprotected |
properties_ | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | private |
save(ModelArchive &zip) const override | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inlinevirtual |
setTraining(bool is_training) | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlinevirtual |
state_map_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | protected |
stateToString() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprotected |
toString() const override | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | inlinevirtual |
training_mode_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | private |
weight_ | Mila::Dnn::Linear< TDeviceType, TInput, TOutput > | private |
~Module()=default | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | virtual |