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. | |