mxnet
Public Member Functions | List of all members
mxnet::common::cuda::DeviceStore Class Reference

#include <cuda_utils.h>

Collaboration diagram for mxnet::common::cuda::DeviceStore:
Collaboration graph

Public Member Functions

 DeviceStore (int requested_device=-1, bool restore=true)
 default constructor- only optionally restores previous device More...
 
 ~DeviceStore ()
 
void SetDevice (int device)
 

Constructor & Destructor Documentation

mxnet::common::cuda::DeviceStore::DeviceStore ( int  requested_device = -1,
bool  restore = true 
)
inlineexplicit

default constructor- only optionally restores previous device

mxnet::common::cuda::DeviceStore::~DeviceStore ( )
inline

Member Function Documentation

void mxnet::common::cuda::DeviceStore::SetDevice ( int  device)
inline

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