ParseSubgraphPostFn
Description
Registers the function for mapping between the input/output nodes in an operator subgraph and those of the operators based on the operator types.
Prototype
APIs using strings will be deprecated in later versions. Use APIs supporting non-string data instead.
OpRegistrationData &ParseSubgraphPostFn(const ParseSubgraphFunc &subgraph_post_fn)
OpRegistrationData &ParseSubgraphPostFn(const ParseSubgraphFuncV2 &subgraph_post_fn)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
subgraph_post_fn |
Input |
Function object for mapping between the input/output nodes in a subgraph and those of the operator. For details, see Callback Function ParseSubgraphFuncV2. |
Restrictions
None
Callback Function ParseSubgraphFuncV2
Call the user-defined ParseSubgraphFuncV2 function to map the input/output nodes in the subgraph to those of the operators.
Status ParseSubgraphFuncV2(const ge::AscendString &subgraph_name, const ge::Graph &graph)
Parameter |
Input/Output |
Description |
|---|---|---|
subgraph_name |
Input |
Subgraph name. |
graph |
Output |
Constructed subgraph. |