|
Mila
Deep Neural Network Library
|

Files | |
| CpuCrossEntropyOp.ixx | |
| Implementation of the CPU-based cross entropy operation for neural networks. | |
| CpuEncoderOp.ixx | |
| Implementation of the CPU-based encoder operation for neural networks. | |
| CpuGeluOp.ixx | |
| CpuLayerNormOp.ixx | |
| Implementation of the CPU-based Layer Normalization operation for neural networks. | |
| CpuLinearOp.ixx | |
| Implementation of the CPU-based Fully Connected operation for neural networks. | |
| CpuMultiHeadAttentionOp.ixx | |
| Implementation of the CPU-based attention operation for neural networks. | |
| CpuOperations.ixx | |
| CpuResidualOp.ixx | |
| Implementation of the CPU-based residual operation for neural networks. | |
| CpuSoftmaxOp.ixx | |
| Implementation of the CPU-based softmax operation for neural networks. | |