This project has retired. For details please refer to its Attic page.
mxnet: mxnet::common::cuda::DeviceStore Class Reference
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 

default constructor- only optionally restores previous device

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

Member Function Documentation

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

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