Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Model< TDeviceType, TPrecision > Member List

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

ensureTrainingMode(const char *method) constMila::Dnn::Model< TDeviceType, TPrecision >inlineprivate
getDeviceId() const noexceptMila::Dnn::Model< TDeviceType, TPrecision >inline
getMemoryStats() constMila::Dnn::Model< TDeviceType, TPrecision >inline
getRuntimeMode() const noexceptMila::Dnn::Model< TDeviceType, TPrecision >inline
isEval() const noexceptMila::Dnn::Model< TDeviceType, TPrecision >inline
isInferenceMode() const noexceptMila::Dnn::Model< TDeviceType, TPrecision >inline
isTrainingMode() const noexceptMila::Dnn::Model< TDeviceType, TPrecision >inline
Model(const Model &)=deleteMila::Dnn::Model< TDeviceType, TPrecision >
Model(Model &&)=defaultMila::Dnn::Model< TDeviceType, TPrecision >
Model(std::unique_ptr< NetworkType > network, RuntimeMode runtime_mode)Mila::Dnn::Model< TDeviceType, TPrecision >inlineexplicitprotected
network_Mila::Dnn::Model< TDeviceType, TPrecision >protected
NetworkType typedefMila::Dnn::Model< TDeviceType, TPrecision >
onTraining()=0Mila::Dnn::Model< TDeviceType, TPrecision >protectedpure virtual
operator=(const Model &)=deleteMila::Dnn::Model< TDeviceType, TPrecision >
operator=(Model &&)=defaultMila::Dnn::Model< TDeviceType, TPrecision >
runtime_mode_Mila::Dnn::Model< TDeviceType, TPrecision >private
setEval(bool eval)Mila::Dnn::Model< TDeviceType, TPrecision >inline
toString() const =0Mila::Dnn::Model< TDeviceType, TPrecision >pure virtual
train()Mila::Dnn::Model< TDeviceType, TPrecision >inline
~Model()=defaultMila::Dnn::Model< TDeviceType, TPrecision >virtual