|
Mila
Deep Neural Network Library
|
Defines the operation types supported by the compute framework. More...
#include <string>#include <stdexcept>Namespaces | |
| namespace | Mila |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
Enumerations | |
| enum class | Mila::Dnn::Compute::OperationType { Mila::Dnn::Compute::CrossEntropyOp , Mila::Dnn::Compute::EncoderOp , Mila::Dnn::Compute::FusedOp , Mila::Dnn::Compute::LinearOp , Mila::Dnn::Compute::GeluOp , Mila::Dnn::Compute::LayerNormOp , Mila::Dnn::Compute::MultiHeadAttentionOp , Mila::Dnn::Compute::ResidualOp , Mila::Dnn::Compute::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. | |
Defines the operation types supported by the compute framework.