注册解析融合算子属性的函数,为FusionParseParamsFn的重载函数。
1 | OpRegistrationData &FusionParseParamsFn(const FusionParseParamByOpFunc &fusion_parse_param_fn)
|
参数 |
输入/输出 |
说明 |
---|---|---|
fusion_parse_param_fn |
输入 |
解析融合算子属性的函数,请参见回调函数FusionParseParamByOpFunc。 |
用户自定义并实现FusionParseParamByOpFunc类函数,完成原始模型中属性到适配昇腾AI处理器的模型中的属性映射,将结果填到Operator类中。
1 | Status FusionParseParamByOpFunc(const std::vector<ge::Operator> &op_src, ge::Operator &op_dest)
|
1 2 3 4 5 | REGISTER_CUSTOM_OP(XXXXXX)
.FrameworkType(TENSORFLOW)
.FusionParseParamsFn(FusionParseParamsFn)
.OriginOpType(XXXXX)
.ImplyType(XXXXX);
|