Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Operations Directory Reference
Directory dependency graph for Operations:
/__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cpu/Operations

Files

 
CpuAttentionOp.ixx
 CPU implementation of Multi-Head Attention operation.
 
CpuCrossEntropyOp.ixx
 Implementation of the CPU-based cross entropy operation for neural networks.
 
CpuEncoderOp.ixx
 CPU backend for the Encoder operation.
 
CpuGeluOp.ixx
 
CpuLayerNormOp.ixx
 CPU implementation of Layer Normalization operation (TensorDataType-based).
 
CpuLinearOp.ixx
 CPU implementation of Linear (fully connected) operation.
 
CpuLinearOpTypeMap.ixx
 LinearOpTraits specialization for CPU / FP32.
 
CpuOperations.ixx
 Aggregated CPU operation module exports.
 
CpuResidualOp.ixx
 CPU implementation of the residual (y = x + F(x)) binary operation.
 
CpuSoftmaxCrossEntropyOp.ixx
 Fused CPU implementation of Softmax + CrossEntropy loss operation.
 
CpuSoftmaxOp.ixx
 CPU implementation of Softmax operation (TensorDataType-based).
 
OperationTraits.Cpu.ixx
 OperationTraits specializations for all CPU operation backends.