单击Gitee或Github,进入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详细了解每一个样例算子,并对样例算子工程进行编译、部署。
不建议对样例工程或自动生成的编译配置文件进行修改,否则可能会造成自定义算子运行失败。
算子名称 |
样例描述 |
---|---|
Add |
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/Add_CN.md”路径中获取详细的样例描述。 |
LeakyRelu |
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/LeakyRelu_CN.md”路径中获取详细的样例描述。 |
算子名称 |
样例描述 |
---|---|
Conv2d |
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/Conv2d_CN.md”路径中获取详细的样例描述。 |
Matmul |
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/Matmul_CN.md”路径中获取详细的样例描述。 |
Permute |
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/Permute_CN.md”路径中获取详细的样例描述。 |
ScatterNdAdd |
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/ScatterNdAdd_CN.md”路径中获取详细的样例描述。 |
Upsample |
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/Upsample_CN.md”路径中获取详细的样例描述。 |
算子名称 |
样例描述 |
---|---|
addblockcust |
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/addblockcust_CN.md”路径中获取详细的样例描述。 |
ReshapeCust |
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/ReshapeCust_CN.md”路径中获取详细的样例描述。 |
UniqueCust |
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“cplusplus/level1_single_api/4_op_dev/1_custom_op/doc/UniqueCust_CN.md”路径中获取详细的样例描述。 |