Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches

Operations → Core Relation

File in Src/Dnn/Compute/Devices/Cuda/OperationsIncludes file in Src/Dnn/Core
Activations / Gelu / CudaGeluOp.Dispatch.ixxComponentConfig.ixx
Activations / Gelu / CudaGeluOp.ixxComponentConfig.ixx
Attention / GQA / CudaGqaOp.ixxComponent.ixx
Attention / GQA / CudaGqaOp.ixxComponentConfig.ixx
Linear / CudaLinearOp.Dispatch.ixxComponentConfig.ixx
Linear / CudaLinearOp.ixxComponent.ixx
Linear / CudaLinearOp.ixxComponentConfig.ixx
Encodings / Lpe / CudaLpeOp.ixxComponent.ixx
Attention / MHA / CudaMhaOp.ixxComponent.ixx
Attention / MHA / CudaMhaOp.ixxComponentConfig.ixx
Loss / CudaSoftmaxCrossEntropyOp.ixxComponent.ixx
Loss / CudaSoftmaxCrossEntropyOp.ixxComponentConfig.ixx
Embeddings / CudaTokenEmbeddingOp.ixxComponent.ixx
Normalizations / LayerNorm / LayerNormOp.ixxComponent.ixx
Normalizations / LayerNorm / LayerNormOp.ixxComponentConfig.ixx
Residual / CudaResidualOp.ixxComponent.ixx
Residual / CudaResidualOp.ixxComponentConfig.ixx
Normalizations / RmsNorm / RmsNormOp.ixxComponent.ixx
Normalizations / RmsNorm / RmsNormOp.ixxComponentConfig.ixx
Encodings / Rope / CudaRopeOp.ixxComponent.ixx
Normalizations / Softmax / CudaSoftmaxOp.ixxComponentConfig.ixx
Normalizations / Softmax / CudaSoftmaxOp.ixxComponent.ixx
Activations / Swiglu / CudaSwigluOp.Dispatch.ixxComponentConfig.ixx
Activations / Swiglu / CudaSwigluOp.ixxComponentConfig.ixx