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::NodePtr &node)
static AGInfo &	Get (const nnvm::NodePtr &node)
static AGInfo &	Create (const nnvm::NodePtr &node)
static bool	IsNone (const NDArray &arr)
static bool	IsVariable (const nnvm::NodePtr &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

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

inline

Member Function Documentation

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

inlinestatic

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

inlinestatic

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

inlinestatic

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

inlinestatic

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

inlinestatic

Member Data Documentation

Context mxnet::Imperative::AGInfo::ctx

bool mxnet::Imperative::AGInfo::fresh_out_grad

OpReqType mxnet::Imperative::AGInfo::grad_req

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

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

OpStatePtr mxnet::Imperative::AGInfo::state

---

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

• include/mxnet/imperative.h