IMPLEMT_COMMON_INFERFUNC

Applicability

Product	Supported or Not
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	√
Atlas inference products	√
Atlas training products	√

Header File

#include <graph/operator_reg.h>

Function Usage

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)

Parameters

Parameter	Input/Output	Description
func_name	Input	User-defined name of the InferShape function.

Returns

None

Constraints

None

Parent topic: Data Structures and APIs