下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

概述

TBE自定义算子开发接口包含算子代码实现接口、插件实现接口以及原型定义接口,如下表所示。

表1 算子开发相关接口汇总

接口类型

章节

功能

算子代码实现接口

TBE DSL API

TBE DSL开发方式相关接口。

TBE TIK API

TBE TIK开发方式相关接口。

TBE DSL Testing API

TBE DSL算子的调试接口。

用于在CPU上对TBE DSL算子进行功能调试。

TBE Platform API

昇腾AI处理器硬件信息公共接口。

TBE Utils API

TBE算子开发过程中的通用校验/操作接口。

TBE Register API

TBE Register模块提供的算子compute函数注册管理等接口。

TBE Context API

TBE Context模块提供的算子编译期间上下文信息管理功能相关接口。

AI CPU API

AI CPU算子功能代码实现相关接口。

原型定义接口

原型定义 API

算子原型定义所需接口。

operator API

原型实现时依赖的operator定义相关接口。

插件适配接口

插件适配 API

算子适配插件实现所需接口。

operator API

插件实现时依赖的operator定义相关接口。

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

当前产品无相关内容

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