Introduction

TBE has encapsulated some verification and operation APIs frequently used in operator development, including the API that checks the operator inputs, outputs, attributes, kernel name, shapes, and data types, as well as the API that broadcasts two inputs and returns the result shape. You can call these APIs directly to facilitate efficient development.

You can find the corresponding API definition files in the CANN component directory. The following table lists the relative paths of the API definition files.

API Category

Header File Path

Operator argument verification

python/site-packages/tbe/common/utils/para_check.py

Shape-related tools

python/site-packages/tbe/common/utils/shape_util.py

Data structs

python/site-packages/tbe/common/utils/para_check.py