Mila
Deep Neural Network Library
|
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. | |