Custom Operator Template

A wealth of custom operator templates included in the CANN package helps you quickly get started.

The custom operator templates are stored in tools/msopgen/template/custom_operator_sample in the CANN component directory.

The custom_operator_sample directory is organized as follows:

├── AICPU
│   ├── Onnx
│   └── Tensorflow
├── DSL
│   ├── Mindspore
│   ├── Onnx
│   ├── PyTorch
│   └── Tensorflow
└── TIK
    ├── Mindspore
    ├── PyTorch
    └── Tensorflow
  • The custom operator template code is used only for operator development reference only. Do not apply the custom operator templates in commercial use.
  • To build an ONNX operator, protoc 3.13.0 is required.