下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

自定义算子开发样例

样例获取

单击GiteeGithub,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op”目录中获取样例。

此样例包含了TBE自定义算子、AI CPU自定义算子以及TensorFlow Scope融合规则开发的代码样例,同时提供了算子编译规则文件。您可以直接基于此样例编译出自定义算子安装包,也可以基于此Sample追加自己的自定义算子实现代码,进行自定义算子的开发。

更多样例请在CANN软件安装后文件存储路径中的“tools/msopgen/template/custom_operator_sample”目录下获取,具体信息请参考自定义算子模板

样例描述

您可以参见样例工程中的README.md详细了解每一个样例算子,并对样例算子工程进行编译、部署。

不建议对样例工程或自动生成的编译配置文件进行修改,否则可能会造成自定义算子运行失败。

表1 AI CPU样例算子列表

算子名称

样例描述

addblockcust

单击GiteeGithub,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/addblockcust_CN.md”路径中获取详细的样例描述。

ReshapeCust

单击GiteeGithub,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/ReshapeCust_CN.md”路径中获取详细的样例描述。

UniqueCust

单击GiteeGithub,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/UniqueCust_CN.md”路径中获取详细的样例描述。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词