mxnet
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...
 
const Contextget_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...
 

Detailed Description

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

Member Function Documentation

const Context& mxnet::RunContext::get_ctx ( ) const
inline

get the base Context from RunContext

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

Context mxnet::RunContext::ctx

base Context

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: