文档
注册

简介

介绍如何使用MindStudio工具实现不同框架的算子开发。

包括支持MindSpore、PyTorch、TensorFlow、Caffe、ONNX的TBE算子开发和支持TensorFlow、PyTorch、ONNX、MindSpore的AI CPU算子开发,其中MindSpore框架的算子目前仅支持Atlas 训练系列产品

本章节主要介绍MindStudio工具操作部分,对代码实现部分仅做基本介绍,如需了解更多代码开发技巧,请参见对应TBE&AI CPU算子开发指南

配置完成Ascend-cann-toolkit开发套件包后,可从Ascend-cann-toolkit安装目录/ascend-toolkit/latest/tools/msopgen/template/operator_demo_projects/获取算子开发样例。样例请参见导入算子工程样例

Ascend-cann-toolkit安装目录/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/impl目录下为昇腾AI处理器支持的内置AI Core算子的实现文件,开发者在进行算子代码实现时可进行参考。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词