Mila
Deep Neural Network Library
|
Implementation of a class to manage device registration. More...
#include <string>
#include <memory>
#include <cuda_runtime.h>
import Compute.CudaDevice;
import Compute.CpuDevice;
import Compute.ComputeDevice;
import Compute.DeviceRegistry;
Classes | |
class | Mila::Dnn::Compute::DeviceRegistrar |
Class to manage compute device initialization. More... | |
Namespaces | |
namespace | Mila |
namespace | Mila::Dnn |
namespace | Mila::Dnn::Compute |
Implementation of a class to manage device registration.