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

Namespace for CUDA fused softmax cross entropy implementation details. More...

Classes
struct	cuda_softmax_crossentropy_impl
	CUDA kernel dispatcher for SoftmaxCrossEntropy operations. More...
struct	cuda_softmax_crossentropy_impl< float >
struct	cuda_softmax_crossentropy_impl< half >

Detailed Description

Namespace for CUDA fused softmax cross entropy implementation details.