Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision > Member List

This is the complete list of members for Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >, including all inherited members.

build(const BuildContext &build_context)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
clearGradients() noexcept	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
data_type (defined in Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	static
DataTypeTraits typedef (defined in Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >
device_type (defined in Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	static
getDataType() const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
getDeviceType() const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
getName() const =0	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	pure virtual
getOperationType() const =0	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	pure virtual
getRequiredStateMemorySize(const BuildContext &build_context) const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
getStateMemorySize() const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
is_built_ (defined in Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	protected
isBuilt() const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
isEvalMode() const	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
setGradients(ITensor *weight_grad, ITensor *bias_grad)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
setParameters(ITensor *weight, ITensor *bias)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
setTrainingMode(TrainingMode training_mode)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	inlinevirtual
training_mode_ (defined in Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	protected
~Operation()=default (defined in Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >)	Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >	virtual