55 const std::vector<std::string>& passes);
101 graph_attr_targets.push_back(attr_name);
111 op_attr_dependency.push_back(attr_name);
121 graph_attr_dependency.push_back(attr_name);
142 #define NNVM_REGISTER_PASS(name) \ 143 DMLC_REGISTRY_REGISTER(::nnvm::PassFunctionReg, PassFunctionReg, name) 147 #endif // NNVM_PASS_H_
Common base class for function registry.
Definition: registry.h:151
PassFunctionReg & set_change_graph(bool v)
Set whether this pass will change graph structure.
Definition: pass.h:90
Graph ApplyPass(Graph src, const std::string &pass)
Apply one pass to the graph.
Definition: pass.h:63
PassFunctionReg & depend_graph_attr(const std::string &attr_name)
Declare this pass requires the given graph attribute to be available before being applied on the grap...
Definition: pass.h:120
std::vector< std::string > op_attr_dependency
dependencies on operator attributes
Definition: pass.h:80
PassFunctionReg & provide_graph_attr(const std::string &attr_name)
Declare that this pass will generate the given graph attribute name once it is applied on the graph...
Definition: pass.h:100
Graph ApplyPasses(Graph src, const std::vector< std::string > &passes)
Apply a series of pass transformations on the input graph.
Symbolic computation graph. This is the intermediate representation for optimization pass...
Definition: graph.h:47
Registry entry for pass functions.
Definition: pass.h:71
std::vector< std::string > graph_attr_dependency
dependencies on attributes in the graph
Definition: pass.h:82
bool change_graph
Whether the pass will change graph structure If this is false, the pass will only change attributes...
Definition: pass.h:78
Configuation of nnvm as well as basic data structure.
PassFunctionReg & depend_op_attr(const std::string &attr_name)
Declare this pass requires the given operator attribute to be available before being applied on the g...
Definition: pass.h:110
std::vector< std::string > graph_attr_targets
generated targets of graph attributes
Definition: pass.h:84
std::function< Graph(Graph src)> PassFunction
A PassFunction is an "Operator on Graph". It takes a source graph and return a graph that may or may ...
Definition: pass.h:46
Configuration of nnvm as well as basic data structure.