Graph

class Graph : public dragon::GraphBase

Graph to execute operators sequentially.

Constructors

dragon::Graph::Graph(
  const GraphDef &def,
  Workspace *ws)

Constructor.

Public Properties

arg

inline const Argument &dragon::Graph::arg(const string &name)

Return the specified argument.

args

inline const Map<string, const Argument*> &dragon::Graph::args()

Return all the arguments.

def

inline const GraphDef &dragon::Graph::def() const

Return the graph def.

name

inline const string &dragon::Graph::name() const

Return the graph name.

optimized_def

inline const GraphDef &dragon::Graph::optimized_def() const

Return the optimized graph def.

phase

inline const string &dragon::Graph::phase() const

Return the executing phase.

workspace

inline Workspace *dragon::Graph::workspace() const

Return the parent workspace.

Public Functions

Create

virtual bool dragon::Graph::Create(const GraphDef &def) override

Create graph in the workspace.

Run

virtual bool dragon::Graph::Run(
  int stream = 0,
  const string &include = "",
  const string &exclude = "") override

Run graph on the given stream.