Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOpRegistrar Class Referenceexport

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: