Mila::Dnn::Compute::CpuLayerNormOp Member List

This is the complete list of members for Mila::Dnn::Compute::CpuLayerNormOp, including all inherited members.

allocateStatisticsTensors()	Mila::Dnn::Compute::CpuLayerNormOp	inlineprivate
asInputTensor(const ITensor &t)	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::FP32 >	inlineprotectedstatic
asOutputTensor(ITensor &t)	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::FP32 >	inlineprotectedstatic
axis_	Mila::Dnn::Compute::CpuLayerNormOp	private
backward(const ITensor &input, const ITensor &output_grad, ITensor &input_grad) const override	Mila::Dnn::Compute::CpuLayerNormOp	inlinevirtual
bias_	Mila::Dnn::Compute::CpuLayerNormOp	private
bias_grad_	Mila::Dnn::Compute::CpuLayerNormOp	private
build(const BuildContext &config) override	Mila::Dnn::Compute::CpuLayerNormOp	inlinevirtual
clearGradients() noexcept	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
computeAxisPartitioning(const shape_t &input_shape)	Mila::Dnn::Compute::CpuLayerNormOp	inlineprivate
config_	Mila::Dnn::Compute::CpuLayerNormOp	private
context_	Mila::Dnn::Compute::CpuLayerNormOp	private
CpuExecutionContext typedef	Mila::Dnn::Compute::CpuLayerNormOp
CpuLayerNormOp(IExecutionContext *context, const LayerNormConfig &config)	Mila::Dnn::Compute::CpuLayerNormOp	inline
data_type	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	static
DataTypeTraits typedef	Mila::Dnn::Compute::Operation< TDeviceType, TInput >
device_type	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	static
dim_size_	Mila::Dnn::Compute::CpuLayerNormOp	private
expected_slices_	Mila::Dnn::Compute::CpuLayerNormOp	private
forward(const ITensor &input, ITensor &output) const override	Mila::Dnn::Compute::CpuLayerNormOp	inlinevirtual
getDataType() const	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
getDeviceType() const	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
getName() const override	Mila::Dnn::Compute::CpuLayerNormOp	inlinevirtual
getOperationType() const override	Mila::Dnn::Compute::CpuLayerNormOp	inlinevirtual
getStateMemorySize() const	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
inner_size_	Mila::Dnn::Compute::CpuLayerNormOp	private
is_built_	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	protected
isBuilt() const	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
isEvalMode() const	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
mean_	Mila::Dnn::Compute::CpuLayerNormOp	private
MR typedef	Mila::Dnn::Compute::CpuLayerNormOp
outer_size_	Mila::Dnn::Compute::CpuLayerNormOp	private
rstd_	Mila::Dnn::Compute::CpuLayerNormOp	private
setGradients(ITensor *weight_grad, ITensor *bias_grad) override	Mila::Dnn::Compute::CpuLayerNormOp	inlinevirtual
setParameters(ITensor *weight, ITensor *bias) override	Mila::Dnn::Compute::CpuLayerNormOp	inlinevirtual
setTrainingMode(TrainingMode training_mode)	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	inlinevirtual
TensorInputType typedef	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::FP32 >
TensorOutputType typedef	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::FP32 >
TensorType typedef	Mila::Dnn::Compute::CpuLayerNormOp
training_mode_	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	protected
UnaryOperationBase typedef	Mila::Dnn::Compute::CpuLayerNormOp
validateInputShape(const shape_t &input_shape) const	Mila::Dnn::Compute::CpuLayerNormOp	inlineprivate
validateShapeConsistency(const shape_t &shape) const	Mila::Dnn::Compute::CpuLayerNormOp	inlineprivate
weight_	Mila::Dnn::Compute::CpuLayerNormOp	private
weight_grad_	Mila::Dnn::Compute::CpuLayerNormOp	private
~Operation()=default	Mila::Dnn::Compute::Operation< TDeviceType, TInput >	virtual
~UnaryOperation()=default	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, TensorDataType::FP32 >	virtual