execution time context. The information needed in runtime for actual execution.
More...
#include <base.h>
|
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.
template<typename xpu >
mshadow::Stream<xpu>* mxnet::RunContext::get_stream |
( |
| ) |
const |
|
inline |
get mshadow stream from Context
- Returns
- the mshadow stream
- Template Parameters
-
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
The documentation for this struct was generated from the following file: