|
Mila
Deep Neural Network Library
|
Enumerations | |
| enum class | Mila::Dnn::Compute::OperationType { CrossEntropyOp , EncoderOp , FusedOp , LinearOp , GeluOp , LayerNormOp , MultiHeadAttentionOp , ResidualOp , SoftmaxOp } |
| Enumeration of all supported neural network operation types. More... | |
Functions | |
| std::string | Mila::Dnn::Compute::operationTypeToString (OperationType op) |
| Converts an operation type to its string representation. | |
Files | |
| file | /home/runner/work/Mila/Mila/Mila/Src/Dnn/Compute/Operations/OperationType.ixx |
| Defines the operation types supported by the compute framework. | |