概述
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