Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Compute.OperationType Module Reference

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.