概述
TBE自定义算子开发接口包含算子代码实现接口、插件实现接口以及原型定义接口,如下表所示。
接口类型  | 
章节  | 
功能  | 
|---|---|---|
算子代码实现接口  | 
TBE DSL开发方式相关接口。  | 
|
TBE TIK开发方式相关接口。  | 
||
TBE DSL算子的调试接口。 用于在CPU上对TBE DSL算子进行功能调试。  | 
||
昇腾AI处理器硬件信息公共接口。  | 
||
TBE算子开发过程中的通用校验/操作接口。  | 
||
TBE Register模块提供的算子compute函数注册管理等接口。  | 
||
TBE Context模块提供的算子编译期间上下文信息管理功能相关接口。  | 
||
AI CPU算子功能代码实现相关接口。  | 
||
原型定义接口  | 
算子原型定义所需接口。  | 
|
原型实现时依赖的operator定义相关接口。  | 
||
插件适配接口  | 
算子适配插件实现所需接口。  | 
|
插件实现时依赖的operator定义相关接口。  | 
父主题: TBE&AI CPU API