自定义算子模板
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。
父主题: 附录