|
Mila
Deep Neural Network Library
|
Exported Modules | |
| module | Dnn.ActivationType |
| module | Compute.MemoryResource |
| module | Dnn.TensorOps |
| module | Dnn.Components.LayerNorm |
| module | Dnn.TensorTypes |
| module | Compute.IExecutionContext |
| module | Dnn.CompositeComponent |
| module | Dnn.ComponentType |
| module | Logging.Logger |
| module | Dnn.Components.Residual |
| module | Compute.ExecutionContextFactory |
| module | Dnn.TensorDataTypeTraits |
| module | Dnn.Components.Linear |
| module | Serialization.ModelArchive |
| module | Dnn.Component |
| module | Compute.CpuMemoryResource |
| module | Dnn.Components.MLP |
| module | Dnn.Components.MultiHeadAttention |
| module | Dnn.TensorDataType |
| module | Serialization.Mode |
| module | Dnn.ITensor |
| module | Compute.ExecutionContext |
| module | Compute.DeviceType |
| module | Dnn.TensorHelpers |
| module | Compute.DeviceTypeTraits |
| module | Compute.Device |
| module | Dnn.Tensor |
| module | Compute.DeviceId |
Classes | |
| class | Mila::Dnn::GptBlock< TDeviceType, TPrecision > |
| Transformer encoder block as a composite component. More... | |
| class | Mila::Dnn::GptBlockConfig |
| Configuration class for GPT transformer blocks. More... | |
Files | |
| file | Mila/Src/Dnn/Components/Transformers/Gpt/GptBlock.ixx |
| Transformer encoder block implementation. | |
| file | Mila/Src/Dnn/Components/Transformers/Gpt/GptBlock.Config.ixx |
| Configuration for GPT-style transformer block (block-level). | |