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