CANN工具提供了自定义算子模板,供用户学习和参考。
自定义算子模板所在路径为:CANN软件安装后文件存储路径中的“tools/msopgen/template/custom_operator_sample”。
custom_operator_sample目录结构。
├── AICPU │ ├── Onnx │ └── Tensorflow ├── DSL │ ├── Mindspore │ ├── Onnx │ ├── PyTorch │ └── Tensorflow └── TIK ├── Mindspore ├── PyTorch └── Tensorflow