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

Namespaces
namespace	Detail
	Namespace for CUDA fused softmax cross entropy implementation details.

Classes
class	CudaSoftmaxCrossEntropyOp
	Fused CUDA implementation of Softmax + CrossEntropy using abstract TensorDataType API. More...
class	CudaSoftmaxCrossEntropyOpRegistrar
	Registrar for fused Softmax+CrossEntropy CUDA operation. More...