mxnet
|
Pass that can be applied to a graph. More...
Go to the source code of this file.
Classes | |
struct | nnvm::PassFunctionReg |
Registry entry for pass functions. More... | |
Namespaces | |
nnvm | |
Macros | |
#define | NNVM_REGISTER_PASS(name) DMLC_REGISTRY_REGISTER(::nnvm::PassFunctionReg, PassFunctionReg, name) |
Macro to register pass fuctions. More... | |
Typedefs | |
typedef std::function< Graph(Graph src)> | nnvm::PassFunction |
A PassFunction is an "Operator on Graph". It takes a source graph and return a graph that may or may not be the same as the input one. More... | |
Functions | |
Graph | nnvm::ApplyPasses (Graph src, const std::vector< std::string > &passes) |
Apply a series of pass transformations on the input graph. More... | |
Graph | nnvm::ApplyPass (Graph src, const std::string &pass) |
Apply one pass to the graph. More... | |
Pass that can be applied to a graph.
#define NNVM_REGISTER_PASS | ( | name | ) | DMLC_REGISTRY_REGISTER(::nnvm::PassFunctionReg, PassFunctionReg, name) |
Macro to register pass fuctions.