文档
注册

简介

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

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词