Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Module< TDeviceType, TInput, TOutput > Member List

This is the complete list of members for Mila::Dnn::Module< TDeviceType, TInput, TOutput >, including all inherited members.

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
getName() constMila::Dnn::Module< TDeviceType, TInput, TOutput >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
isTraining() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inline
load(ModelArchive &archive)=0Mila::Dnn::Module< TDeviceType, TInput, TOutput >pure virtual
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
MR typedefMila::Dnn::Module< TDeviceType, TInput, TOutput >
operator<<Mila::Dnn::Module< TDeviceType, TInput, TOutput >friend
parameter_map_Mila::Dnn::Module< TDeviceType, TInput, TOutput >protected
parameterCount() const =0Mila::Dnn::Module< TDeviceType, TInput, TOutput >pure virtual
parametersToString() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inlineprotected
save(ModelArchive &archive) const =0Mila::Dnn::Module< TDeviceType, TInput, TOutput >pure virtual
setTraining(bool is_training)Mila::Dnn::Module< TDeviceType, TInput, TOutput >inlinevirtual
state_map_Mila::Dnn::Module< TDeviceType, TInput, TOutput >protected
stateToString() constMila::Dnn::Module< TDeviceType, TInput, TOutput >inlineprotected
toString() const =0Mila::Dnn::Module< TDeviceType, TInput, TOutput >pure virtual
training_mode_Mila::Dnn::Module< TDeviceType, TInput, TOutput >private
~Module()=defaultMila::Dnn::Module< TDeviceType, TInput, TOutput >virtual