|
Mila 0.13.48
Deep Neural Network Library
|
Aggregated CUDA operation module exports. More...
import Compute.CudaResidualOp;import Compute.CudaGeluOp;import Compute.CudaSoftmaxOp;import Compute.CudaRopeOp;import Compute.CudaTokenEmbeddingOp;import Compute.CudaLinearOp;import Compute.CudaRmsNormOp;import Compute.CudaLayerNormOp;import Compute.CudaGqaOp;import Compute.CudaSwigluOp;import Compute.CudaLpeOp;import Compute.CudaMultiHeadAttentionOp;Aggregated CUDA operation module exports.
This module re-exports CUDA-specific operation implementations so callers can import a single module that exposes the available CUDA operation modules and their factory registrations.