|
Mila
Deep Neural Network Library
|
Device type definitions and conversion utilities for compute devices. More...
#include <stdexcept>#include <algorithm>#include <string>Namespaces | |
| namespace | Mila |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
Enumerations | |
| enum class | Mila::Dnn::Compute::DeviceType { Mila::Dnn::Compute::Cpu , Mila::Dnn::Compute::Cuda } |
| Enumeration of supported compute device types. More... | |
Functions | |
| std::string | Mila::Dnn::Compute::deviceToString (DeviceType device_type) |
| Converts a DeviceType to its string representation. | |
| DeviceType | Mila::Dnn::Compute::toDeviceType (std::string device_type) |
| Converts a string to the corresponding DeviceType. | |
Device type definitions and conversion utilities for compute devices.