Skip to content

概述

本章节内容介绍AI框架调用自定义算子的方法。如下图所示,Pytorch支持单算子和图模式两种,TensorFlow仅支持图模式。

AI框架调用时,除了需要提供CANN框架调用时需要的代码实现文件,还需要进行插件适配开发。

免责声明:本站内容由 asc-devkit 仓 master 分支自动编译生成,属于持续开发版本,可能存在缺陷,仅供预览与参考。如需稳定及商用资料,请查阅官方 昇腾社区