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

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

addComponent(ComponentPtr component)	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inline
backward(const TensorType &input, const TensorType &output_grad)	Mila::Dnn::MLP< TDeviceType, TPrecision >	inline
build(const BuildContext &context) final	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
build_context_	Mila::Dnn::Component< TDeviceType, TPrecision >	protected
childCount() const noexcept	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inline
clearComponents()	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inline
Component(const std::string &name)	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineexplicit
ComponentBase typedef (defined in Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >)	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
ComponentPtr typedef (defined in Mila::Dnn::MLP< TDeviceType, TPrecision >)	Mila::Dnn::MLP< TDeviceType, TPrecision >
CompositeComponent(const std::string &name)	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inlineexplicit
CompositeComponent(const CompositeComponent &)=delete (defined in Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >)	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
CompositeComponent(CompositeComponent &&) noexcept=default (defined in Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >)	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
CompositeComponentBase typedef (defined in Mila::Dnn::MLP< TDeviceType, TPrecision >)	Mila::Dnn::MLP< TDeviceType, TPrecision >
decode(const TensorType &input) const	Mila::Dnn::MLP< TDeviceType, TPrecision >	inline
findComponent(const std::string &path) const	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inline
forward(const TensorType &input)	Mila::Dnn::MLP< TDeviceType, TPrecision >	inline
GeluType typedef (defined in Mila::Dnn::MLP< TDeviceType, TPrecision >)	Mila::Dnn::MLP< TDeviceType, TPrecision >
getComponent(const std::string &name) const	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inline
getComponentAs(const std::string &name) const	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inlineprotected
getComponents() const	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inline
getDeviceId() const override	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inlinevirtual
getDeviceType()	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinestatic
getExecutionContext() const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlineprotected
getGradients() const override	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inlinevirtual
getMemoryStats() const override	Mila::Dnn::MLP< TDeviceType, TPrecision >	inlinevirtual
getName() const	Mila::Dnn::Component< TDeviceType, TPrecision >	inline
getParameterNames() const	Mila::Dnn::Component< TDeviceType, TPrecision >	inlinevirtual
getParameters() const override	Mila::Dnn::CompositeComponent< 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::MLP< TDeviceType, TPrecision >	inlinevirtual
hasChildren() const noexcept	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inline
hasComponent(const std::string &name) const	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inline
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
LinearType typedef (defined in Mila::Dnn::MLP< TDeviceType, TPrecision >)	Mila::Dnn::MLP< TDeviceType, TPrecision >
load_(ModelArchive &archive, SerializationMode mode) override	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inlineprotectedvirtual
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
MLP(const std::string &name, const MLPConfig &config, std::optional< DeviceId > device_id=std::nullopt)	Mila::Dnn::MLP< TDeviceType, TPrecision >	inlineexplicit
MR typedef (defined in Mila::Dnn::MLP< TDeviceType, TPrecision >)	Mila::Dnn::MLP< TDeviceType, TPrecision >
onBuilding(const BuildContext &context) override	Mila::Dnn::MLP< TDeviceType, TPrecision >	inlineprotectedvirtual
onExecutionContextSet() override	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inlineprotectedvirtual
onTrainingModeChanging(TrainingMode training_mode) override	Mila::Dnn::MLP< TDeviceType, TPrecision >	inlineprotectedvirtual
operator=(const CompositeComponent &)=delete (defined in Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >)	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
operator=(CompositeComponent &&) noexcept=default (defined in Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >)	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
optimize()	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inlineprotectedvirtual
parameterCount() const override	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inlinevirtual
removeComponent(const std::string &name)	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inline
requireSerializableParameters() const override	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inlineprotectedvirtual
save_(ModelArchive &archive, SerializationMode mode) const override (defined in Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >)	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inlineprotectedvirtual
saveFlatTensors(Serialization::SafeTensorsWriter &writer, const std::string &prefix, Serialization::TensorSavePass pass) const override	Mila::Dnn::CompositeComponent< 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::CompositeComponent< TDeviceType, TPrecision >	inlinevirtual
TensorType typedef (defined in Mila::Dnn::MLP< TDeviceType, TPrecision >)	Mila::Dnn::MLP< TDeviceType, TPrecision >
toString() const override	Mila::Dnn::MLP< TDeviceType, TPrecision >	inlinevirtual
tryFindComponent(const std::string &path) const	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	inline
zeroGradients() override	Mila::Dnn::MLP< TDeviceType, TPrecision >	inlinevirtual
~Component()=default (defined in Mila::Dnn::Component< TDeviceType, TPrecision >)	Mila::Dnn::Component< TDeviceType, TPrecision >	virtual
~CompositeComponent()=default (defined in Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >)	Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >	virtual
~MLP() override=default	Mila::Dnn::MLP< TDeviceType, TPrecision >