COMMON_INFER_FUNC_REG
Description
Registers the InferShape function of an operator.
Different from INFER_FUNC_REG, the argument of the InferShape function registered by this function is of the base class instead of the subclass of Operator. This API allows the InferShape function to be reused across operators.
Prototype
COMMON_INFER_FUNC_REG(op_name, x)
This API call automatically issues a COMMON_INFER_VERIFY_FUNC(x) call, in which x indicates the pointer to x passed to the COMMON_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_COMMON_INFERFUNC. |
Returns
None
Parent topic: GE Namespace