注册解析算子属性的函数。
1 | OpRegistrationData &ParseParamsFn(const ParseParamFunc &parseParamFn)
|
参数 |
输入/输出 |
说明 |
---|---|---|
parseParamFn |
输入 |
解析算子属性的函数ParseParamFunc,请参见回调函数ParseParamFunc。 针对TensorFlow框架,若原始TensorFlow框架算子属性与适配昇腾AI处理器的模型中算子属性一一对应(属性个数与顺序一致),可直接使用AutoMappingFn函数自动实现映射。 |
对于自定义算子插件,ParseParamsFn后续版本将会废弃,请使用ParseParamsByOperatorFn接口进行算子属性的解析。
若用户已使用ParseParamsFn接口进行了算子插件的开发,请执行如下操作进行新接口适配:
用户自定义并实现FusionParseParamFunc类函数,完成原始模型中算子属性到适配昇腾AI处理器的模型中算子属性映射,将结果填到Operator类中。
1 | using ParseParamFunc = std::function<domi::Status(const google::protobuf::Message *, ge::Operator &)>
|