Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Compute.CudaLayerNormOp Module Reference

Classes

class  Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >
 CUDA implementation of Layer Normalization. More...
class  Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOpRegistrar

Files

file  /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Operations/Normalizations/LayerNorm/LayerNormOp.ixx
file  /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Operations/Normalizations/LayerNorm/LayerNormOp.Dispatch.ixx