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 |
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 |
config_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | private |
createContext(const std::string &device_name) | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inlineprivatestatic |
device_context_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | private |
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 |
isTraining() const | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | inline |
load(ModelArchive &archive) override | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | 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::CompositeModule< TDeviceType, TDataType > | |
MR typedef | Mila::Dnn::CompositeModule< TDeviceType, TDataType > | |
parameter_map_ | Mila::Dnn::Module< TDeviceType, TInput, TOutput > | protected |
parameterCount() const override | Mila::Dnn::CompositeModule< 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 |
save(ModelArchive &archive) const override | Mila::Dnn::CompositeModule< 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::CompositeModule< 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 |