Symbolic computation graph. This is the intermediate representation for optimization pass. More...
#include <graph.h>
Public Member Functions | |
| template<typename T > | |
| const T & | GetAttr (const std::string &attr_name) const |
| Get the immutable attribute from attrs. More... | |
| bool | HasAttr (const std::string &attr_name) const |
| Check whether has a specific attribute. More... | |
| template<typename T > | |
| T | MoveCopyAttr (const std::string &attr_name) |
| Get a move copy of the attribute, implement copy on write semantics. The content is moved if the reference counter of shared_ptr is 1. The attribute is erased from attrs after the call. More... | |
| const IndexedGraph & | indexed_graph () const |
| get a indexed graph of current graph, if not exist, create it on demand More... | |
Public Attributes | |
| std::vector< NodeEntry > | outputs |
| outputs of the computation graph. More... | |
| std::unordered_map< std::string, std::shared_ptr< any > > | attrs |
| attributes of a graph Note that attribute is shared pointer and can be shared across graphs. More... | |
Detailed Description
Symbolic computation graph. This is the intermediate representation for optimization pass.
Member Function Documentation
◆ GetAttr()
|
inline |
Get the immutable attribute from attrs.
- Parameters
-
attr_name the name of the attribute
- Returns
- the reference to corresponding attribute
- Template Parameters
-
T the type of the attribute.
◆ HasAttr()
|
inline |
Check whether has a specific attribute.
- Parameters
-
attr_name the name of the attribute
- Returns
- a boolean result
◆ indexed_graph()
| const IndexedGraph& nnvm::Graph::indexed_graph | ( | ) | const |
get a indexed graph of current graph, if not exist, create it on demand
- Returns
- The indexed graph.
- See also
- IndexedGraph
◆ MoveCopyAttr()
|
inline |
Get a move copy of the attribute, implement copy on write semantics. The content is moved if the reference counter of shared_ptr is 1. The attribute is erased from attrs after the call.
- Parameters
-
attr_name the name of the attribute
- Returns
- a new copy of the corresponding attribute.
- Template Parameters
-
T the type of the attribute.
Member Data Documentation
◆ attrs
| std::unordered_map<std::string, std::shared_ptr<any> > nnvm::Graph::attrs |
attributes of a graph Note that attribute is shared pointer and can be shared across graphs.
It is highly recommended to keep each attribute immutable. It is also safe to implement an copy-on-write semnatics.
Copy when shared_ptr.unique is not true, while reuse original space when shared_ptr.unique is true.
◆ outputs
| std::vector<NodeEntry> nnvm::Graph::outputs |
outputs of the computation graph.
The documentation for this class was generated from the following file:
- /work/mxnet/3rdparty/tvm/nnvm/include/nnvm/graph.h
