|
Mila
Deep Neural Network Library
|
Classes | |
| class | Mila::Dnn::CompositeModule< TDeviceType, TDataType > |
| A module class that can contain and manage child modules. More... | |
Typedefs | |
| template<typename TDataType = float> | |
| using | Mila::Dnn::CpuCompositeModule = CompositeModule< DeviceType::Cpu, TDataType > |
| template<typename TDataType = float> | |
| using | Mila::Dnn::CudaCompositeModule = CompositeModule< DeviceType::Cuda, TDataType > |
Files | |
| file | /home/runner/work/Mila/Mila/Mila/Src/Dnn/Modules/CompositeModule.ixx |
| Implementation of the composite module pattern for neural network components. | |