IMPLEMT_COMMON_INFERFUNC
Description
Encapsulates the Common_InferShape function of an operator.
Compared with IMPLEMT_INFERFUNC, this function automatically generates an op object of class Operator, which can directly call Operator to implement InferShape. If InferShape is universal and can be called by the prototypes of multiple operators, you can use this API.
Prototype
IMPLEMT_COMMON_INFERFUNC(func_name)
Restrictions
None
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
func_name |
Input |
User-defined name of the InferShape function. |
Returns
None
Parent topic: GE Namespace