Mila::Dnn::Compute::CpuSoftmaxCrossEntropyOpRegistrar Class Reference

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

Static Public Member Functions
static void	registerOperations ()

Static Public Attributes
static bool	isRegistered

Detailed Description

Registrar for fused Softmax+CrossEntropy operation.

Member Function Documentation

registerOperations()

void Mila::Dnn::Compute::CpuSoftmaxCrossEntropyOpRegistrar::registerOperations ( )

inlinestatic

Here is the call graph for this function:

Member Data Documentation

isRegistered

bool Mila::Dnn::Compute::CpuSoftmaxCrossEntropyOpRegistrar::isRegistered

inlinestatic

Initial value:

= []() {
            registerOperations();
            return true;
            }()

---

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

• /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cpu/Operations/CpuSoftmaxCrossEntropyOp.ixx