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

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

Static Public Member Functions

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

Detailed Description

Class responsible for registering the CudaLinearOp operation.

The CudaLinearOpRegistrar class registers the CudaLinearOp operation with the OperationRegistry. It associates the operation name "Cuda::LinearOp" with a factory function that creates instances of CudaLinearOp.

Member Function Documentation

◆ registerOperations()

static void Mila::Dnn::Compute::CudaLinearOpRegistrar::registerOperations ( )
inlinestatic

Registers the CudaLinearOp operation with the OperationRegistry.

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

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: