nnvm::Node Class Reference

Node represents an operation in a computation graph. More...

#include <node.h>

Collaboration diagram for nnvm::Node:

Public Member Functions
	Node ()=default
	Node (const Op *op, const std::string &name)
	~Node ()
	destructor of node More...
const Op *	op () const
bool	is_variable () const
	return whether node is placeholder variable. This is equivalent to op == nullptr More...
uint32_t	num_outputs () const
uint32_t	num_inputs () const

Static Public Member Functions
template<class... Args>
static NodePtr	Create (Args &&...args)
	create a new empty shared_ptr of Node. More...

Public Attributes
NodeAttrs	attrs
	The attributes in the node. More...
std::vector< NodeEntry >	inputs
	inputs to this node More...
std::vector< NodePtr >	control_deps
	Optional control flow dependencies Gives operation must be performed before this operation. More...
any	info
	additional fields for this node More...

Detailed Description

Node represents an operation in a computation graph.

Constructor & Destructor Documentation

nnvm::Node::Node ( )

default

nnvm::Node::Node ( const Op *  op, const std::string &  name  )

inline

nnvm::Node::~Node

(

)

destructor of node

Member Function Documentation

template<class... Args>

static NodePtr nnvm::Node::Create ( Args &&...  args )

inlinestatic

create a new empty shared_ptr of Node.

Returns

a created empty node.

bool nnvm::Node::is_variable ( ) const

inline

return whether node is placeholder variable. This is equivalent to op == nullptr

Returns

whether node is placeholder input variable

uint32_t nnvm::Node::num_inputs ( ) const

inline

Returns

number of inputs from this node

uint32_t nnvm::Node::num_outputs ( ) const

inline

Returns

number of outputs from this node

const Op * nnvm::Node::op ( ) const

inline

Returns

operator in this node

Member Data Documentation

NodeAttrs nnvm::Node::attrs

The attributes in the node.

std::vector<NodePtr> nnvm::Node::control_deps

Optional control flow dependencies Gives operation must be performed before this operation.

any nnvm::Node::info

additional fields for this node

std::vector<NodeEntry> nnvm::Node::inputs

inputs to this node

---

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

• /work/mxnet/3rdparty/tvm/nnvm/include/nnvm/node.h