Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Compute.DeviceType Module Reference

Enumerations

enum class  Mila::Dnn::Compute::DeviceType { Cpu , 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.
 

Files

file  /home/runner/work/Mila/Mila/Mila/Src/Dnn/Compute/Devices/DeviceType.ixx
 Device type definitions and conversion utilities for compute devices.