mxnet
|
Registry entry for pass functions. More...
#include <pass.h>
Public Member Functions | |
PassFunctionReg & | set_change_graph (bool v) |
Set whether this pass will change graph structure. More... | |
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. More... | |
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 graph. More... | |
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 graph. More... | |
Public Member Functions inherited from dmlc::FunctionRegEntryBase< PassFunctionReg, PassFunction > | |
PassFunctionReg & | set_body (PassFunctionbody) |
Set the function body. More... | |
PassFunctionReg & | describe (const std::string &description) |
Describe the function. More... | |
PassFunctionReg & | add_argument (const std::string &name, const std::string &type, const std::string &description) |
Add argument information to the function. More... | |
PassFunctionReg & | add_arguments (const std::vector< ParamFieldInfo > &args) |
Append list if arguments to the end. More... | |
PassFunctionReg & | set_return_type (const std::string &type) |
Set the return type. More... | |
Public Attributes | |
bool | change_graph {false} |
Whether the pass will change graph structure If this is false, the pass will only change attributes. More... | |
std::vector< std::string > | op_attr_dependency |
dependencies on operator attributes More... | |
std::vector< std::string > | graph_attr_dependency |
dependencies on attributes in the graph More... | |
std::vector< std::string > | graph_attr_targets |
generated targets of graph attributes More... | |
Public Attributes inherited from dmlc::FunctionRegEntryBase< PassFunctionReg, PassFunction > | |
std::string | name |
name of the entry More... | |
std::string | description |
description of the entry More... | |
std::vector< ParamFieldInfo > | arguments |
additional arguments to the factory function More... | |
PassFunction | body |
Function body to create ProductType. More... | |
std::string | return_type |
Return type of the function. More... | |
Additional Inherited Members | |
Protected Member Functions inherited from dmlc::FunctionRegEntryBase< PassFunctionReg, PassFunction > | |
PassFunctionReg & | self () |
Registry entry for pass functions.
|
inline |
Declare this pass requires the given graph attribute to be available before being applied on the graph.
attr_name | Name of the attribute. |
|
inline |
Declare this pass requires the given operator attribute to be available before being applied on the graph.
attr_name | Name of the attribute. |
|
inline |
Declare that this pass will generate the given graph attribute name once it is applied on the graph.
attr_name | Name of the graph attribute. |
|
inline |
Set whether this pass will change graph structure.
v | If true, the pass will change graph structure. |
bool nnvm::PassFunctionReg::change_graph {false} |
Whether the pass will change graph structure If this is false, the pass will only change attributes.
std::vector<std::string> nnvm::PassFunctionReg::graph_attr_dependency |
dependencies on attributes in the graph
std::vector<std::string> nnvm::PassFunctionReg::graph_attr_targets |
generated targets of graph attributes
std::vector<std::string> nnvm::PassFunctionReg::op_attr_dependency |
dependencies on operator attributes