Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::CompositeModule< TDeviceType, TDataType > Member List

This is the complete list of members for Mila::Dnn::CompositeModule< TDeviceType, TDataType >, including all inherited members.

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() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getDeviceType() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getModule(const std::string &name) constMila::Dnn::CompositeModule< TDeviceType, TDataType >inline
getModules() constMila::Dnn::CompositeModule< TDeviceType, TDataType >inline
getName() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getNamedModules() constMila::Dnn::CompositeModule< TDeviceType, TDataType >inline
getParameterTensors() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getPrecision() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
getStateTensors() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
hasModule(const std::string &name) constMila::Dnn::CompositeModule< TDeviceType, TDataType >inline
isTraining() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
load(ModelArchive &archive) overrideMila::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 typedefMila::Dnn::CompositeModule< TDeviceType, TDataType >
MR typedefMila::Dnn::CompositeModule< TDeviceType, TDataType >
parameter_map_Mila::Dnn::Module< TDeviceType, TInput, TOutput >protected
parameterCount() const overrideMila::Dnn::CompositeModule< TDeviceType, TDataType >inlinevirtual
parametersToString() constMila::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 overrideMila::Dnn::CompositeModule< TDeviceType, TDataType >inlinevirtual
setTraining(bool is_training) overrideMila::Dnn::CompositeModule< TDeviceType, TDataType >inlinevirtual
state_map_Mila::Dnn::Module< TDeviceType, TInput, TOutput >protected
stateToString() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inlineprotected
toString() const overrideMila::Dnn::CompositeModule< TDeviceType, TDataType >inlinevirtual
training_mode_Mila::Dnn::Module< TDeviceType, TInput, TOutput >private
~CompositeModule()=defaultMila::Dnn::CompositeModule< TDeviceType, TDataType >virtual
~Module()=defaultMila::Dnn::Module< TDeviceType, TInput, TOutput >virtual