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

Class responsible for registering the CudaGeluOp operation. More...

Static Public Member Functions

static void registerOperations ()
 Registers the CudaGeluOp operation with the OperationRegistry.

Detailed Description

Class responsible for registering the CudaGeluOp operation.

The CudaGeluOpRegistrar class registers the CudaGeluOp operation with the OperationRegistry. It associates the operation name "Cuda::GeluOp" with a factory function that creates instances of CudaGeluOp.

Member Function Documentation

◆ registerOperations()

void Mila::Dnn::Compute::Cuda::Gelu::CudaGeluOpRegistrar::registerOperations ( )
inlinestatic

Registers the CudaGeluOp operation with the OperationRegistry.

This function registers the CudaGeluOp operation for the CUDA device type with the OperationRegistry. It associates the operation name "Cuda::GeluOp" with a factory function that creates instances of CudaGeluOp.

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/Cuda/Operations/Activations/Gelu/CudaGeluOp.ixx