mxnet
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Public Member Functions | Public Attributes | List of all members
mxnet::RunContext Struct Reference

execution time context. The information needed in runtime for actual execution. More...

#include <base.h>

Collaboration diagram for mxnet::RunContext:
Collaboration graph

Public Member Functions

template<typename xpu >
mshadow::Stream< xpu > * get_stream () const
 get mshadow stream from Context More...
 

Public Attributes

void * stream
 the stream of the device, can be NULL or Stream<gpu>* in GPU mode More...
 

Detailed Description

execution time context. The information needed in runtime for actual execution.

Member Function Documentation

template<typename xpu >
mshadow::Stream<xpu>* mxnet::RunContext::get_stream ( ) const
inline

get mshadow stream from Context

Returns
the mshadow stream
Template Parameters
xputhe device type of the stream

Member Data Documentation

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: