ParseSubgraphPostFn
函数功能
根据算子类型,注册算子的子图中输入输出节点跟算子的输入输出的对应关系函数实现。
函数原型

数据类型为string的接口后续版本会废弃,建议使用数据类型为非string的接口。
OpRegistrationData &ParseSubgraphPostFn(const ParseSubgraphFunc &subgraph_post_fn)
OpRegistrationData &ParseSubgraphPostFn(const ParseSubgraphFuncV2 &subgraph_post_fn);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
subgraph_post_fn |
输入 |
子图中输入输出节点跟算子的输入输出的对应关系函数对象。 |
约束说明
无。
父主题: OpRegistrationData类