|
Mila 0.13.48
Deep Neural Network Library
|
Exported Modules | |
| module | Compute.IExecutionContext |
| module | Compute.DeviceType |
| module | Compute.ExecutionContextTemplate |
| module | Compute.DeviceId |
Typedefs | |
| using | Mila::Dnn::Compute::CpuExecutionContext = ExecutionContext<DeviceType::Cpu> |
Functions | |
| template<DeviceType TDeviceType> | |
| const ExecutionContext< TDeviceType > * | Mila::Dnn::Compute::cast_context_ (const IExecutionContext *ctx) noexcept |
| template<DeviceType TDeviceType> | |
| ExecutionContext< TDeviceType > * | Mila::Dnn::Compute::cast_context_ (IExecutionContext *ctx) noexcept |
| template<DeviceType TDeviceType> | |
| ExecutionContext< TDeviceType > * | Mila::Dnn::Compute::validateExecutionContext_ (IExecutionContext *context, const std::string &op_name) |
Files | |
| file | /__w/Mila/Mila/Mila/Src/Dnn/Compute/ExecutionContext.ixx |
| Templated execution context framework for compute operations and stream management. | |
| file | /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cpu/CpuExecutionContext.ixx |
| CPU-specific execution context specialization. | |
| file | /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/CudaExecutionContext.ixx |
| CUDA-specific execution context specialization. | |