我要评分

文档获取效率

文档正确性

内容完整性

文档易理解

在线提单

论坛求助

昇腾小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算子功能代码实现相关接口。