自定义算子开发
样例获取
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op”目录中获取样例。
此样例包含了TBE自定义算子、AI CPU自定义算子以及TensorFlow Scope融合规则开发的代码样例,同时提供了算子编译规则文件。开发者可直接基于此样例编译出自定义算子安装包。
您也可以基于此样例追加自己的自定义算子实现代码,进行自定义算子的开发。
样例描述
您可以参见样例工程中的README.md详细了解每一个样例算子,并对样例算子工程进行编译、部署。
父主题: 样例参考