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.