mxnet
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
mxnet::Imperative::AGInfo Class Reference

#include <imperative.h>

Collaboration diagram for mxnet::Imperative::AGInfo:
Collaboration graph

Public Member Functions

 AGInfo ()
 

Static Public Member Functions

static void Clear (const nnvm::NodePtr &node)
 
static AGInfoGet (const nnvm::NodePtr &node)
 
static AGInfoCreate (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< NDArrayoutputs
 
std::vector< NDArrayout_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: