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.