三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

CANN元数据融合解析函数

CANN元数据融合解析函数

FusionParseParamsFn(Overload)

【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef

函数功能

注册解析融合算子属性的函数,为FusionParseParamsFn的重载函数。

函数原型

[OpRegistrationData](https://link.gitcode.com/i/e6632fb1f94c5561e1fe38517fc4adc3) &FusionParseParamsFn(const FusionParseParamByOpFunc &fusion_parse_param_fn)

参数说明

参数

输入/输出

说明

fusion_parse_param_fn

输入

解析融合算子属性的函数,请参见回调函数FusionParseParamByOpFunc。

回调函数FusionParseParamByOpFunc

用户自定义并实现FusionParseParamByOpFunc类函数,完成原始模型中属性到适配昇腾AI处理器的模型中的属性映射,将结果填到Operator类中。

Status FusionParseParamByOpFunc(const std::vector<ge::Operator> &op_src, ge::Operator &op_dest)

表 1参数说明

参数

输入/输出

说明

op_src

输入

一组scope内存储原始模型中算子属性的融合算子数据结构。

op_dest

输出

融合算子数据结构,保存融合算子信息。

调用示例

REGISTER_CUSTOM_OP(XXXXXX) .FrameworkType(TENSORFLOW) .FusionParseParamsFn(FusionParseParamsFn) .OriginOpType(XXXXX) .ImplyType(XXXXX);

【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

← 返回列表