Compute.CudaSoftmaxCrossEntropyOp Module Reference

Classes
class	Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets >
	Fused CUDA implementation of Softmax + CrossEntropy using abstract TensorDataType API. More...
class	Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOpRegistrar
	Registrar for fused Softmax+CrossEntropy CUDA operation. More...

Files
file	/__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Operations/Loss/CudaSoftmaxCrossEntropyOp.ixx
	Fused CUDA implementation of Softmax + CrossEntropy loss operation.