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