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

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

build(const BuildContext &context) finalMila::Dnn::Component< TDeviceType, TPrecision >inlinevirtual
build_context_Mila::Dnn::Component< TDeviceType, TPrecision >protected
built_Mila::Dnn::Component< TDeviceType, TPrecision >private
Component(const std::string &name)Mila::Dnn::Component< TDeviceType, TPrecision >inlineexplicit
ensureBuilt(const char *method) constMila::Dnn::Component< TDeviceType, TPrecision >inlineprivate
exec_context_Mila::Dnn::Component< TDeviceType, TPrecision >private
ExecutionContextType typedefMila::Dnn::Loss< TDeviceType, TPrecision >
getDeviceId() const =0Mila::Dnn::Component< TDeviceType, TPrecision >pure virtual
getDeviceType()Mila::Dnn::Component< TDeviceType, TPrecision >inlinestatic
getExecutionContext() constMila::Dnn::Component< TDeviceType, TPrecision >inlineprotected
getGradients() const =0Mila::Dnn::Component< TDeviceType, TPrecision >pure virtual
getMemoryStats() const =0Mila::Dnn::Component< TDeviceType, TPrecision >pure virtual
getName() constMila::Dnn::Component< TDeviceType, TPrecision >inline
getParameterNames() constMila::Dnn::Component< TDeviceType, TPrecision >inlinevirtual
getParameters() const =0Mila::Dnn::Component< TDeviceType, TPrecision >pure virtual
getPrecision() noexceptMila::Dnn::Component< TDeviceType, TPrecision >inlinestatic
getRuntimeMode() const noexceptMila::Dnn::Component< TDeviceType, TPrecision >inline
getTrainingMode() const noexceptMila::Dnn::Component< TDeviceType, TPrecision >inline
getType() const =0Mila::Dnn::Component< TDeviceType, TPrecision >pure virtual
hasExecutionContext() const noexceptMila::Dnn::Component< TDeviceType, TPrecision >inlineprotected
isBuilt() const finalMila::Dnn::Component< TDeviceType, TPrecision >inlinevirtual
isIdentifier(const std::string &s) noexceptMila::Dnn::Component< TDeviceType, TPrecision >inlineprivatestatic
isInferenceMode() const noexceptMila::Dnn::Component< TDeviceType, TPrecision >inline
isTrainingMode() const noexceptMila::Dnn::Component< TDeviceType, TPrecision >inline
loadParameter(const std::string &name, const Serialization::ITensorBlob &blob)Mila::Dnn::Component< TDeviceType, TPrecision >inlinevirtual
loadParameterFromBlob(const std::string &param_name, const Serialization::ITensorBlob &blob, Tensor< TParameterPrecision, TMemoryResource > &target, const shape_t &expected_shape)Mila::Dnn::Component< TDeviceType, TPrecision >inlineprotected
MR typedefMila::Dnn::Loss< TDeviceType, TPrecision >
name_Mila::Dnn::Component< TDeviceType, TPrecision >private
onBuilding(const BuildContext &config)Mila::Dnn::Component< TDeviceType, TPrecision >inlineprotectedvirtual
onExecutionContextSet()Mila::Dnn::Component< TDeviceType, TPrecision >inlineprotectedvirtual
onTrainingModeChanging(TrainingMode mode)Mila::Dnn::Component< TDeviceType, TPrecision >inlineprotectedvirtual
parameterCount() const =0Mila::Dnn::Component< TDeviceType, TPrecision >pure virtual
save_(ModelArchive &archive, SerializationMode mode) const =0Mila::Dnn::Component< TDeviceType, TPrecision >pure virtual
setExecutionContext(IExecutionContext *context)Mila::Dnn::Component< TDeviceType, TPrecision >inlineprotected
setTrainingMode(TrainingMode mode)Mila::Dnn::Component< TDeviceType, TPrecision >inline
synchronize()=0Mila::Dnn::Component< TDeviceType, TPrecision >pure virtual
TensorType typedefMila::Dnn::Loss< TDeviceType, TPrecision >
toString() const =0Mila::Dnn::Component< TDeviceType, TPrecision >pure virtual
training_mode_Mila::Dnn::Component< TDeviceType, TPrecision >private
training_mode_mutex_Mila::Dnn::Component< TDeviceType, TPrecision >private
validateName(const std::string &name)Mila::Dnn::Component< TDeviceType, TPrecision >inlineprivatestatic
zeroGradients()Mila::Dnn::Component< TDeviceType, TPrecision >inlinevirtual
~Component()=defaultMila::Dnn::Component< TDeviceType, TPrecision >virtual
~Loss()=defaultMila::Dnn::Loss< TDeviceType, TPrecision >virtual