|
Mila 0.13.48
Deep Neural Network Library
|

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. | |