简介
TBE封装了一些算子开发过程中的通用校验/操作接口,例如对算子输入/输出、属性、Kernel Name进行基础校验,shape与data type的校验,对两个输入进行broadcast操作并返回broadcast后的shape等。开发者在算子开发过程中可直接调用这些接口,提升算子开发效率。
您可以在CANN软件安装后文件存储路径下找到对应接口定义文件,接口定义文件相对路径如下表所示。
接口分类 |
头文件路径 |
---|---|
算子参数校验 |
python/site-packages/tbe/common/utils/para_check.py |
shape相关工具 |
python/site-packages/tbe/common/utils/shape_util.py |
数据结构定义 |
python/site-packages/tbe/common/utils/para_check.py |
父主题: TBE Utils API