act_output_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
activation_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
addModule(const std::string &name, std::shared_ptr< Module< TDeviceType, TDataType, TDataType > > module) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
addModule(std::shared_ptr< Module< TDeviceType, TDataType, TDataType > > module) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
backward(const Tensor< TDataType, MR > &input, const Tensor< TDataType, MR > &output_grad, Tensor< TDataType, MR > &input_grad) | Mila::Dnn::MLP< TDeviceType, TDataType > | inline |
child_module_map_ | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | private |
child_modules_ | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | private |
CompositeModule() | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
CompositeModule(const std::string &device_name, const ComponentConfig &config) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inlineexplicit |
CompositeModule(std::shared_ptr< DeviceContext > context, const ComponentConfig &config) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inlineexplicit |
CompositeModuleBase typedef | Mila::Dnn::MLP< TDeviceType, TDataType > | |
config_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
createContext(const std::string &device_name) | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprivatestatic |
device_context_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | private |
dropout1_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
dropout1_output_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
fc1_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
fc1_output_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
fc2_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
fc2_output_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
forward(const Tensor< TDataType, MR > &input, Tensor< TDataType, MR > &output) | Mila::Dnn::MLP< TDeviceType, TDataType > | inline |
getDeviceContext() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
getDeviceType() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
getModule(const std::string &name) const | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
getModules() const | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
getName() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
getNamedModules() const | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | 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 |
hasModule(const std::string &name) const | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
initializeModules() | Mila::Dnn::MLP< TDeviceType, TDataType > | inlineprivate |
isTraining() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
load(ModelArchive &archive) override | Mila::Dnn::MLP< TDeviceType, TDataType > | inlinevirtual |
MLP(const std::string &device_name, const MLPConfig &config) | Mila::Dnn::MLP< TDeviceType, TDataType > | inlineexplicit |
MLP(std::shared_ptr< DeviceContext > device_context, const MLPConfig &config) | Mila::Dnn::MLP< TDeviceType, TDataType > | inlineexplicit |
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::CompositeModule< TDeviceType, TDataType > | |
MR typedef | Mila::Dnn::MLP< TDeviceType, TDataType > | |
norm1_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
norm1_output_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
parameter_map_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | protected |
parameterCount() const override | Mila::Dnn::MLP< TDeviceType, TDataType > | inlinevirtual |
parametersToString() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprotected |
removeModule(const std::string &name) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
replaceModule(const std::string &name, std::shared_ptr< Module< TDeviceType, TDataType, TDataType > > module) | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inline |
residual_input_ | Mila::Dnn::MLP< TDeviceType, TDataType > | private |
save(ModelArchive &archive) const override | Mila::Dnn::MLP< TDeviceType, TDataType > | inlinevirtual |
setTraining(bool is_training) override | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | inlinevirtual |
state_map_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | protected |
stateToString() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprotected |
toString() const override | Mila::Dnn::MLP< TDeviceType, TDataType > | inlinevirtual |
training_mode_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | private |
~CompositeModule()=default | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | virtual |
~Module()=default | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | virtual |