INFER_FUNC_REG
Applicable Products
All processors
Header File
#include <graph/operator_reg.h>
Function Usage
Registers the InferShape function of an operator.
Prototype
1 | INFER_FUNC_REG(op_name, x) |
This API call automatically issues an INFER_VERIFY_FUNC(op_name, x) call, in which op_name indicates the operator type, and x indicates the pointer to x passed to the INFER_FUNC_REG(op_name,x) call.
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
op_name |
Input |
Operator type. |
x |
Input |
InferShape function name, which must be the same as the InferShape function name passed to IMPLEMT_INFERFUNC. |
Returns
None
Restrictions
None
Parent topic: Basic Data Structures and APIs of Graphs