自定义算子模板
CANN工具提供了自定义算子模板,供用户学习和参考。
自定义算子模板所在路径为:CANN软件安装后文件存储路径中的“tools/msopgen/template/custom_operator_sample”。
custom_operator_sample目录结构。
├── AICPU │ ├── Onnx │ └── Tensorflow ├── DSL │ ├── Mindspore │ ├── Onnx │ ├── PyTorch │ └── Tensorflow └── TIK ├── Mindspore ├── PyTorch └── Tensorflow
- 自定义算子模板代码仅作为示例,供算子开发过程参考,不支持用户基于自定义算子模板进行产品化开发。
- Onnx算子编译时需要protoc版本为3.13.0。
父主题: 附录