mxnet::ext::Graph Class Reference

#include <lib_api.h>

Collaboration diagram for mxnet::ext::Graph:

Public Member Functions
	Graph ()
	~Graph ()
JsonVal	toJson () const
std::string	toString () const
void	_dfs_util (Node *n, std::unordered_set< Node *> *to_visit, std::function< void(Node *)> handler) const
void	DFS (std::function< void(Node *)> handler) const
std::vector< Node * >	topological_sort () const
void	print (int indent=0) const
Node *	addNode (const std::string &name, const std::string &op)
Node *	getNode (size_t idx)
const Node *	getNode (size_t idx) const
const JsonVal &	getAttr (const std::string &key) const
size_t	size () const
void	_setPassResource (PassResource *res_)
void	_setParams (std::unordered_map< std::string, mxnet::ext::MXTensor > *args, std::unordered_map< std::string, mxnet::ext::MXTensor > *aux)

Static Public Member Functions
static Graph *	fromString (const std::string &json)
static Graph *	fromJson (JsonVal val)

Public Attributes
std::vector< Node * >	inputs
std::vector< NodeEntry >	outputs
std::map< std::string, JsonVal >	attrs

Constructor & Destructor Documentation

Graph()

mxnet::ext::Graph::Graph

(

)

~Graph()

mxnet::ext::Graph::~Graph

(

)

Member Function Documentation

_dfs_util()

void mxnet::ext::Graph::_dfs_util	(	Node *	n,
		std::unordered_set< Node *> *	to_visit,
		std::function< void(Node *)>	handler
	)		const

_setParams()

void mxnet::ext::Graph::_setParams	(	std::unordered_map< std::string, mxnet::ext::MXTensor > *	args,
		std::unordered_map< std::string, mxnet::ext::MXTensor > *	aux
	)

_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()

static Graph* mxnet::ext::Graph::fromJson ( JsonVal  val )

static

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

Member Data Documentation

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:

• /work/mxnet/include/mxnet/lib_api.h