| asInputTensor(const ITensor &t) | Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TPrecision > | inlineprotectedstatic |
| asOutputTensor(ITensor &t) | Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TPrecision > | inlineprotectedstatic |
| backward(const ITensor &input, const ITensor &output_grad, ITensor &input_grad) const override | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inlinevirtual |
| bias_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| bias_grad_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| build(const BuildContext &config) override | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inlinevirtual |
| clearGradients() noexcept | Mila::Dnn::Compute::Operation< TDeviceType, TInput > | inlinevirtual |
| computeRuntimePartition_(const shape_t &input_shape, int64_t &norm_axis, int &outer_size, int &inner_size, int &norm_dim, int64_t &num_slices, int64_t &normalized_features) const | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inlineprivate |
| config_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| context_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| CudaExecutionContext typedef | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | |
| CudaLayerNormOp(IExecutionContext *context, const LayerNormConfig &config) | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | 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 |
| forward(const ITensor &input, ITensor &output) const override | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inlinevirtual |
| getConfig() const | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inline |
| getDataType() const | Mila::Dnn::Compute::Operation< TDeviceType, TInput > | inlinevirtual |
| getDeviceType() const | Mila::Dnn::Compute::Operation< TDeviceType, TInput > | inlinevirtual |
| getName() const override | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inlinevirtual |
| getOperationType() const override | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inlinevirtual |
| getStateMemorySize() const | Mila::Dnn::Compute::Operation< TDeviceType, TInput > | inlinevirtual |
| impl_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | 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 |
| max_inner_size_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| max_input_shape_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| max_norm_dim_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| max_num_slices_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| max_outer_size_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| mean_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| mean_tensor_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| MR typedef | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | |
| NativeType typedef | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | |
| norm_axis_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| rstd_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| rstd_tensor_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| setGradients(ITensor *weight_grad, ITensor *bias_grad) override | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inlinevirtual |
| setParameters(ITensor *weight, ITensor *bias) override | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inlinevirtual |
| setTrainingMode(TrainingMode training_mode) | Mila::Dnn::Compute::Operation< TDeviceType, TInput > | inlinevirtual |
| TensorInputType typedef | Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TPrecision > | |
| TensorOutputType typedef | Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TPrecision > | |
| TensorType typedef | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | |
| training_mode_ | Mila::Dnn::Compute::Operation< TDeviceType, TInput > | protected |
| UnaryOperationBase typedef | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | |
| validateNormalizedShape_(const shape_t &input_shape) const | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inlineprivate |
| validateRuntimeShape_(const shape_t &input_shape) const | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | inlineprivate |
| weight_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| weight_grad_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| weight_size_ | Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision > | private |
| ~Operation()=default | Mila::Dnn::Compute::Operation< TDeviceType, TInput > | virtual |
| ~UnaryOperation()=default | Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TPrecision > | virtual |
1.15.0