Compute.FusedSoftmaxCrossEntropyOp Module Reference

Classes
class	Mila::Dnn::Compute::FusedSoftmaxCrossEntropyOp< TPrecision >
	CUDA implementation of the fused softmax and cross entropy operation for neural networks. More...
class	Mila::Dnn::Compute::FusedSoftmaxCrossEntropyOpRegistrar
	Class responsible for registering the FusedSoftmaxCrossEntropyOp operation. More...

Files
file	/home/runner/work/Mila/Mila/Mila/Src/Dnn/Compute/Operations/Cuda/FusedSoftmaxCrossEntropyOp.ixx
	Implementation of the CUDA-based fused softmax and cross entropy operation for neural networks.