#include <lib_api.h>
◆ Graph()
mxnet::ext::Graph::Graph |
( |
| ) |
|
◆ ~Graph()
mxnet::ext::Graph::~Graph |
( |
| ) |
|
◆ _dfs_util()
void mxnet::ext::Graph::_dfs_util |
( |
Node * |
n, |
|
|
std::unordered_set< Node *> * |
to_visit, |
|
|
std::function< void(Node *)> |
handler |
|
) |
| const |
◆ _setParams()
◆ _setPassResource()
void mxnet::ext::Graph::_setPassResource |
( |
PassResource * |
res_ | ) |
|
◆ addNode()
Node* mxnet::ext::Graph::addNode |
( |
const std::string & |
name, |
|
|
const std::string & |
op |
|
) |
| |
◆ DFS()
void mxnet::ext::Graph::DFS |
( |
std::function< void(Node *)> |
handler | ) |
const |
◆ fromJson()
◆ fromString()
static Graph* mxnet::ext::Graph::fromString |
( |
const std::string & |
json | ) |
|
|
static |
◆ getAttr()
const JsonVal& mxnet::ext::Graph::getAttr |
( |
const std::string & |
key | ) |
const |
◆ getNode() [1/2]
Node* mxnet::ext::Graph::getNode |
( |
size_t |
idx | ) |
|
◆ getNode() [2/2]
const Node* mxnet::ext::Graph::getNode |
( |
size_t |
idx | ) |
const |
◆ print()
void mxnet::ext::Graph::print |
( |
int |
indent = 0 | ) |
const |
◆ size()
size_t mxnet::ext::Graph::size |
( |
| ) |
const |
◆ toJson()
JsonVal mxnet::ext::Graph::toJson |
( |
| ) |
const |
◆ topological_sort()
std::vector<Node*> mxnet::ext::Graph::topological_sort |
( |
| ) |
const |
◆ toString()
std::string mxnet::ext::Graph::toString |
( |
| ) |
const |
◆ attrs
std::map<std::string, JsonVal> mxnet::ext::Graph::attrs |
◆ inputs
std::vector<Node*> mxnet::ext::Graph::inputs |
◆ outputs
std::vector<NodeEntry> mxnet::ext::Graph::outputs |
The documentation for this class was generated from the following file: