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

Class responsible for registering CPU GELU operations. More...

Static Public Member Functions

static void registerOperations ()
 Registers CPU GELU operations with the OperationRegistry.

Detailed Description

Class responsible for registering CPU GELU operations.

Registers CPU GELU operation implementations with the OperationRegistry for different precisions. Currently supports FP32.

The registrar uses static initialization to automatically register operations when the module is loaded.

Member Function Documentation

◆ registerOperations()

void Mila::Dnn::Compute::CpuGeluOpRegistrar::registerOperations ( )
inlinestatic

Registers CPU GELU operations with the OperationRegistry.

Registers factory functions for creating CPU GELU operations with different data types. Currently registers:

Here is the call graph for this function:
Here is the caller graph for this function:

The documentation for this class was generated from the following file:
  • /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cpu/Operations/CpuGeluOp.ixx