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

Enumerations

enum class  Mila::Dnn::ComponentType : int {
  Unknown = 0 , Linear , Gelu , Swiglu ,
  LayerNorm , RmsNorm , Softmax , Dropout ,
  MultiHeadAttention , GroupedQueryAttention , Residual , TokenEmbedding ,
  Lpe , Rope , SoftmaxCrossEntropy , Mlp ,
  Transformer , Network , Gpt2 , Llama ,
  Mistral , Bert , 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  /__w/Mila/Mila/Mila/Src/Dnn/Core/ComponentType.ixx
 Enumeration of built-in component types supported by the deserializer.