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 |