mxnet
|
Context info passing from MXNet OpContext dev_type is string repr of supported context, currently only "cpu" and "gpu" dev_id is the device index where the tensor locates. More...
#include <lib_api.h>
Public Member Functions | |
MXContext () | |
MXContext (std::string dev_type_, int dev_id_) | |
MXContext (const char *dev_type_, int dev_id_) | |
Static Public Member Functions | |
static MXContext | CPU () |
static MXContext | GPU () |
static MXContext | CPU (int dev_id) |
static MXContext | GPU (int dev_id) |
Public Attributes | |
std::string | dev_type |
int | dev_id |
Context info passing from MXNet OpContext dev_type is string repr of supported context, currently only "cpu" and "gpu" dev_id is the device index where the tensor locates.
|
inline |
|
inlineexplicit |
|
inlineexplicit |
|
inlinestatic |
|
inlinestatic |
|
inlinestatic |
|
inlinestatic |
int MXContext::dev_id |
std::string MXContext::dev_type |