Mila::Dnn::FusedModule< TPrecision, TInput, TDeviceType > Member List

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

build(Device, DType) override	Mila::Dnn::FusedModule< TPrecision, TInput, TDeviceType >	inline
config_	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	private
createContext(const std::string &device_name)	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inlineprivatestatic
device_context_	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	private
forward(const Tensor &input, Tensor &output) override	Mila::Dnn::FusedModule< TPrecision, TInput, TDeviceType >	inline
FusedModule(std::shared_ptr< FusedOp > op)	Mila::Dnn::FusedModule< TPrecision, TInput, TDeviceType >	inlineexplicit
getDeviceContext() const	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inline
getDeviceType() const	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inline
getName() const	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inline
getParameterTensors() const	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inline
getPrecision() const	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inline
getStateTensors() const	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inline
isTraining() const	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inline
load(ModelArchive &archive)=0	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	pure virtual
Module(const std::string &device_name, const ComponentConfig &config)	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inlineexplicit
Module(std::shared_ptr< DeviceContext > context, const ComponentConfig &config)	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inlineexplicit
MR typedef	Mila::Dnn::Module< TDeviceType, TInput, TOutput >
op_	Mila::Dnn::FusedModule< TPrecision, TInput, TDeviceType >	private
parameter_map_	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	protected
parameterCount() const =0	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	pure virtual
parametersToString() const	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inlineprotected
save(ModelArchive &archive) const =0	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	pure virtual
setTraining(bool is_training)	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inlinevirtual
state_map_	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	protected
stateToString() const	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	inlineprotected
toString() const =0	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	pure virtual
training_mode_	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	private
~Module()=default	Mila::Dnn::Module< TDeviceType, TInput, TOutput >	virtual