Node represents an operation in a computation graph.
More...
#include <node.h>
Node represents an operation in a computation graph.
◆ Node() [1/2]
◆ Node() [2/2]
nnvm::Node::Node |
( |
const Op * |
op, |
|
|
const std::string & |
name |
|
) |
| |
|
inline |
◆ ~Node()
◆ Create()
template<class ... Args>
static ObjectPtr nnvm::Node::Create |
( |
Args &&... |
args | ) |
|
|
inlinestatic |
create a new empty shared_ptr of Node.
- Returns
- a created empty node.
◆ is_variable()
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
◆ num_inputs()
uint32_t nnvm::Node::num_inputs |
( |
| ) |
const |
|
inline |
- Returns
- number of inputs from this node
◆ num_outputs()
uint32_t nnvm::Node::num_outputs |
( |
| ) |
const |
|
inline |
- Returns
- number of outputs from this node
◆ op()
const Op * nnvm::Node::op |
( |
| ) |
const |
|
inline |
- Returns
- operator in this node
◆ attrs
The attributes in the node.
◆ control_deps
std::vector<ObjectPtr> nnvm::Node::control_deps |
Optional control flow dependencies Gives operation must be performed before this operation.
◆ info
additional fields for this node
◆ inputs
The documentation for this class was generated from the following file:
- /work/mxnet/3rdparty/tvm/nnvm/include/nnvm/node.h