mxnet::Imperative::AGInfo Class Reference

#include <imperative.h>

Collaboration diagram for mxnet::Imperative::AGInfo:

Public Member Functions
	AGInfo ()

Static Public Member Functions
static void	Clear (const nnvm::ObjectPtr &node)
static AGInfo &	Get (const nnvm::ObjectPtr &node)
static AGInfo &	Create (const nnvm::ObjectPtr &node)
static bool	IsNone (const NDArray &arr)
static bool	IsVariable (const nnvm::ObjectPtr &node)

Public Attributes
Context	ctx
OpReqType	grad_req
OpStatePtr	state
std::vector< NDArray >	outputs
std::vector< NDArray >	out_grads
bool	fresh_out_grad

Constructor & Destructor Documentation

AGInfo()

mxnet::Imperative::AGInfo::AGInfo ( )

inline

Member Function Documentation

Clear()

static void mxnet::Imperative::AGInfo::Clear ( const nnvm::ObjectPtr &  node )

inlinestatic

Create()

static AGInfo& mxnet::Imperative::AGInfo::Create ( const nnvm::ObjectPtr &  node )

inlinestatic

Get()

static AGInfo& mxnet::Imperative::AGInfo::Get ( const nnvm::ObjectPtr &  node )

inlinestatic

IsNone()

static bool mxnet::Imperative::AGInfo::IsNone ( const NDArray &  arr )

inlinestatic

IsVariable()

static bool mxnet::Imperative::AGInfo::IsVariable ( const nnvm::ObjectPtr &  node )

inlinestatic

Member Data Documentation

ctx

Context mxnet::Imperative::AGInfo::ctx

fresh_out_grad

bool mxnet::Imperative::AGInfo::fresh_out_grad

grad_req

OpReqType mxnet::Imperative::AGInfo::grad_req

out_grads

std::vector<NDArray> mxnet::Imperative::AGInfo::out_grads

outputs

std::vector<NDArray> mxnet::Imperative::AGInfo::outputs

state

OpStatePtr mxnet::Imperative::AGInfo::state

---

The documentation for this class was generated from the following file:

• /work/mxnet/include/mxnet/imperative.h