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