Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Compute.CudaMatMulBiasGeluOp Module Reference

Classes

class  Mila::Dnn::Compute::CudaMatMulBiasGeluOp< TInput, TOutput >
 CUDA implementation of the fused MatMul-Bias-GELU operation. More...
 
class  Mila::Dnn::Compute::CudaMatMulBiasGeluOpRegistrar
 Class responsible for registering the CudaMatMulBiasGeluOp operation. More...
 

Files

file  /home/runner/work/Mila/Mila/Mila/Src/Dnn/Compute/Operations/Cuda/MatMulBiasActivation.ixx
 Implementation of fused matrix multiplication, bias addition, and activation operations.