Compute.CudaLayerNormOp Module Reference

Classes
class	Mila::Dnn::Compute::CudaLayerNormOp< TInput, TOutput >
	CUDA implementation of the Layer Normalization operation for neural networks. More...
class	Mila::Dnn::Compute::CudaLayerNormOpRegistrar
	Class responsible for registering the CudaLayerNormOp operation. More...

Files
file	/home/runner/work/Mila/Mila/Mila/Src/Dnn/Compute/Operations/Cuda/CudaLayerNormOp.ixx
	Implementation of the CUDA-based Layer Normalization operation for neural networks.