Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Dnn.ComponentType Module Reference

Enumerations

enum class  Mila::Dnn::ComponentType : int {
  Unknown = 0 , Linear , Gelu , Activation ,
  Swiglu , LayerNorm , RmsNorm , Softmax ,
  Dropout , MultiHeadAttention , GroupedQueryAttention , Residual ,
  TokenEmbedding , Lpe , Rope , SoftmaxCrossEntropy ,
  Mlp , GatedMlp , Transformer , Network ,
  CustomComponentStart = 1000 , MockComponent = CustomComponentStart
}
 Canonical list of framework-known component types. More...

Functions

ComponentType Mila::Dnn::fromString (std::string_view s) noexcept
 Parse a case-insensitive component name into a ComponentType.
ComponentType Mila::Dnn::fromTypeId (std::string_view s) noexcept
 Map a short type identifier back to a ComponentType enum.
std::string Mila::Dnn::toString (ComponentType t) noexcept
 Convert a ComponentType enum value to its canonical name.
std::string Mila::Dnn::toTypeId (ComponentType t) noexcept
 Get the short 2..4 character type identifier for a ComponentType.

Files

file  Mila/Src/Dnn/Core/ComponentType.ixx
 Enumeration of built-in component types supported by the deserializer.