INFER_FUNC_REG
Description
Registers the InferShape function of an operator.
Prototype
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.
Restrictions
None
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
Parent topic: GE Namespace