LayerNormOp.Dispatch.ixx File Reference

#include <cublasLt.h>
#include <cuda_fp16.h>
#include <vector>
#include <memory>
#include <string>
#include <stdexcept>
#include <cstdint>
#include <type_traits>
#include <sstream>
#include <cassert>
#include "Kernels/LayerNorm.cuh"

Classes
struct	Mila::Dnn::Compute::Cuda::LayerNorm::Detail::cuda_layernorm_impl< float >
struct	Mila::Dnn::Compute::Cuda::LayerNorm::Detail::cuda_layernorm_impl< half >

Namespaces
namespace	Mila
	Mila main API namespace.
namespace	Mila::Dnn
namespace	Mila::Dnn::Compute
namespace	Mila::Dnn::Compute::Cuda
namespace	Mila::Dnn::Compute::Cuda::LayerNorm
namespace	Mila::Dnn::Compute::Cuda::LayerNorm::Detail