Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOpRegistrar Class Reference

Registrar for fused Softmax+CrossEntropy CUDA operation. More...

Static Public Member Functions
static void	registerOperations ()

Detailed Description

Registrar for fused Softmax+CrossEntropy CUDA operation.

Member Function Documentation

registerOperations()

void Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOpRegistrar::registerOperations ( )

inlinestatic

Here is the call graph for this function:

---

The documentation for this class was generated from the following file:

• /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Operations/Loss/CudaSoftmaxCrossEntropyOp.ixx