Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
OperationType.ixx File Reference

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.
 

Detailed Description

Defines the operation types supported by the compute framework.