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

Classes
struct	cuda_gelu_impl
struct	cuda_gelu_impl< float >
struct	cuda_gelu_impl< half >

Typedefs
using	BackwardFp16Func = void (*)(half*, const half*, const half*, int, cudaStream_t)
using	BackwardFp32Func = void (*)(float*, const float*, const float*, int, cudaStream_t)
using	ForwardFp16Func = void (*)(half*, const half*, int, cudaStream_t)
using	ForwardFp32Func = void (*)(float*, const float*, int, cudaStream_t)

Typedef Documentation

BackwardFp16Func

using Mila::Dnn::Compute::Cuda::Gelu::Detail::BackwardFp16Func = void (*)(half*, const half*, const half*, int, cudaStream_t)

BackwardFp32Func

using Mila::Dnn::Compute::Cuda::Gelu::Detail::BackwardFp32Func = void (*)(float*, const float*, const float*, int, cudaStream_t)

ForwardFp16Func

using Mila::Dnn::Compute::Cuda::Gelu::Detail::ForwardFp16Func = void (*)(half*, const half*, int, cudaStream_t)

ForwardFp32Func

using Mila::Dnn::Compute::Cuda::Gelu::Detail::ForwardFp32Func = void (*)(float*, const float*, int, cudaStream_t)