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