#include <cuda_utils.h>

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

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: