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

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

backward(const TensorType &input, const TensorType &output_grad)	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inline
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
ComponentBase typedef (defined in Mila::Dnn::LayerNorm< TDeviceType, TPrecision >)	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >
forward(const TensorType &input)	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inline
getDeviceId() const override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
getDeviceType()	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinestatic
getExecutionContext() const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineprotected
getGradients() const override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
getMemoryStats() const override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
getName() const	Mila::Dnn::Component< TDeviceType, TPrecision >	inline
getParameterNames() const override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
getParameters() const override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
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 override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
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
LayerNorm(const std::string &name, const LayerNormConfig &config, std::optional< DeviceId > device_id=std::nullopt)	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlineexplicit
load_(ModelArchive &archive, SerializationMode mode)	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
loadParameter(const std::string &name, const ITensorBlob &blob) override	Mila::Dnn::LayerNorm< 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 typedef (defined in Mila::Dnn::LayerNorm< TDeviceType, TPrecision >)	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >
onBuilding(const BuildContext &context) override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlineprotectedvirtual
onExecutionContextSet() override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlineprotectedvirtual
onTrainingModeChanging(TrainingMode training_mode) override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlineprotectedvirtual
parameterCount() const override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
requireSerializableParameters() const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
save_(ModelArchive &archive, SerializationMode mode) const override (defined in Mila::Dnn::LayerNorm< TDeviceType, TPrecision >)	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
saveFlatTensors(Serialization::SafeTensorsWriter &writer, const std::string &prefix, Serialization::TensorSavePass pass) const override	Mila::Dnn::LayerNorm< 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() override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
TensorType typedef (defined in Mila::Dnn::LayerNorm< TDeviceType, TPrecision >)	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >
toString() const override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
zeroGradients() override	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >	inlinevirtual
~Component()=default (defined in Mila::Dnn::Component< TDeviceType, TPrecision >)	Mila::Dnn::Component< TDeviceType, TPrecision >	virtual
~LayerNorm() override=default (defined in Mila::Dnn::LayerNorm< TDeviceType, TPrecision >)	Mila::Dnn::LayerNorm< TDeviceType, TPrecision >