注册解析算子属性的函数。
OpRegistrationData &ParseParamsFn(const ParseParamFunc& parseParamFn)
参数 |
输入/输出 |
说明 |
|---|---|---|
parseParamFn |
输入 |
解析算子属性的函数ParseParamFunc，请参见回调函数ParseParamFunc。 针对TensorFlow框架，若原始TensorFlow框架算子属性与适配昇腾AI处理器的模型中算子属性一一对应（属性个数与顺序一致），可直接使用AutoMappingFn函数自动实现映射。 |
对于自定义算子插件，ParseParamsFn后续版本将会废弃，请使用ParseParamsByOperatorFn接口进行算子属性的解析。
若用户已使用ParseParamsFn接口进行了算子插件的开发，请执行如下操作进行新接口适配：
用户自定义并实现FusionParseParamFunc类函数，完成原始模型中算子属性到适配昇腾AI处理器的模型中算子属性映射，将结果填到Operator类中。
Status ParseParamFunc(const Message* op_origin, ge::Operator& op_dest)
参数 |
输入/输出 |
说明 |
|---|---|---|
op_origin |
输入 |
protobuf格式的数据结构（来源于原始模型的prototxt文件），包含算子属性信息。 |
op_dest |
输出 |
适配昇腾AI处理器的模型的算子数据结构，保存算子信息。 关于Operator类，请参见Operator类。 |