Overview

This section describes how to use MindStudio to develop custom operators under different frameworks.

For example, develop TBE operators under MindSpore, PyTorch, TensorFlow, Caffe, and ONNX frameworks and develop AI CPU operators under MindSpore, TensorFlow, PyTorch, and ONNX frameworks. Currently, MindSpore operators apply only to Atlas training productss.

This section mainly focuses on MindStudio operations but also touches on important code implementations. For details about the programming, see the TBE & AI CPU Operator Developer Guide.

After the Ascend-CANN-Toolkit configuration is complete, you can obtain an operator development sample from Ascend-CANN-Toolkit_installation_directory/ascend-toolkit/latest/tools/msopgen/template/operator_demo_projects. For details, see Importing an Operator Project.

The Ascend-CANN-Toolkit_installation_directory/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/impl directory stores the implementation files of the built-in AI Core operators supported by the Ascend AI Processor. You can refer to these files when implementing the operator code.