Mila::Dnn::Compute::Cuda::LayerNorm::Detail Namespace Reference

Classes
struct	cuda_layernorm_impl
	CUDA kernel dispatcher for LayerNorm operations. More...
struct	cuda_layernorm_impl< float >
struct	cuda_layernorm_impl< half >