Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Public Member Functions | List of all members
dragon::CUDADeviceGuard Class Reference

#include <cuda_device.h>

Public Member Functions

 CUDADeviceGuard (int new_id)
 
 ~CUDADeviceGuard ()
 

Constructor & Destructor Documentation

◆ CUDADeviceGuard()

dragon::CUDADeviceGuard::CUDADeviceGuard ( int  new_id)
inline

◆ ~CUDADeviceGuard()

dragon::CUDADeviceGuard::~CUDADeviceGuard ( )
inline

The documentation for this class was generated from the following file: