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

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.