Mila::Dnn::Component< TDeviceType, TPrecision > Member List

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

build(const BuildContext &context) final	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
build_context_	Mila::Dnn::Component< TDeviceType, TPrecision >	protected
Component(const std::string &name)	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineexplicit
CompositeComponent (defined in Mila::Dnn::Component< TDeviceType, TPrecision >)	Mila::Dnn::Component< TDeviceType, TPrecision >	friend
getDeviceId() const =0	Mila::Dnn::Component< TDeviceType, TPrecision >	pure virtual
getDeviceType()	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinestatic
getExecutionContext() const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineprotected
getGradients() const =0	Mila::Dnn::Component< TDeviceType, TPrecision >	pure virtual
getMemoryStats() const =0	Mila::Dnn::Component< TDeviceType, TPrecision >	pure virtual
getName() const	Mila::Dnn::Component< TDeviceType, TPrecision >	inline
getParameterNames() const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
getParameters() const =0	Mila::Dnn::Component< TDeviceType, TPrecision >	pure virtual
getPrecision() noexcept	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinestatic
getRequiredMemory(const BuildContext &context) const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
getTrainingMode() const noexcept	Mila::Dnn::Component< TDeviceType, TPrecision >	inline
getType() const =0	Mila::Dnn::Component< TDeviceType, TPrecision >	pure virtual
hasExecutionContext() const noexcept	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineprotected
HostStagingMemoryResource typedef	Mila::Dnn::Component< TDeviceType, TPrecision >	protected
isBuilt() const final	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
load_(ModelArchive &archive, SerializationMode mode)	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
loadParameter(const std::string &, const Serialization::ITensorBlob &)	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
Network (defined in Mila::Dnn::Component< TDeviceType, TPrecision >)	Mila::Dnn::Component< TDeviceType, TPrecision >	friend
onBuilding(const BuildContext &)	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineprotectedvirtual
onExecutionContextSet()	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineprotectedvirtual
onTrainingModeChanging(TrainingMode)	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineprotectedvirtual
operator<<(std::ostream &os, const Component &component)	Mila::Dnn::Component< TDeviceType, TPrecision >	friend
parameterCount() const =0	Mila::Dnn::Component< TDeviceType, TPrecision >	pure virtual
requireSerializableParameters() const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
save_(ModelArchive &archive, SerializationMode mode) const =0 (defined in Mila::Dnn::Component< TDeviceType, TPrecision >)	Mila::Dnn::Component< TDeviceType, TPrecision >	pure virtual
saveFlatTensors(Serialization::SafeTensorsWriter &writer, const std::string &prefix, Serialization::TensorSavePass pass) const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
saveParameterToArchive(ModelArchive &archive, const std::string &parameter_name, const Tensor< TParameterPrecision, TMemoryResource > &parameter) const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineprotected
saveParameterToWriter(Serialization::SafeTensorsWriter &writer, const std::string &flat_name, const Tensor< TParameterPrecision, TMemoryResource > &parameter, Serialization::TensorSavePass pass) const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineprotected
setExecutionContext(IExecutionContext *context)	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineprotected
setTrainingMode(TrainingMode mode)	Mila::Dnn::Component< TDeviceType, TPrecision >	inline
synchronize()=0	Mila::Dnn::Component< TDeviceType, TPrecision >	pure virtual
toString() const =0	Mila::Dnn::Component< TDeviceType, TPrecision >	pure virtual
zeroGradients()	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
~Component()=default (defined in Mila::Dnn::Component< TDeviceType, TPrecision >)	Mila::Dnn::Component< TDeviceType, TPrecision >	virtual