mxnet
|
execution time context. The information needed in runtime for actual execution. More...
#include <base.h>
Public Member Functions | |
template<typename xpu > | |
mshadow::Stream< xpu > * | get_stream () const |
get mshadow stream from Context More... | |
const Context & | get_ctx () const |
get the base Context from RunContext More... | |
Public Attributes | |
Context | ctx |
base Context More... | |
void * | stream |
the stream of the device, can be NULL or Stream<gpu>* in GPU mode More... | |
execution time context. The information needed in runtime for actual execution.
|
inline |
get the base Context from RunContext
|
inline |
get mshadow stream from Context
xpu | the device type of the stream |
void* mxnet::RunContext::stream |
the stream of the device, can be NULL or Stream<gpu>* in GPU mode