IMPLEMT_COMMON_INFERFUNC

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

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

Restrictions

None

Parent topic: Basic Data Structures and APIs of Graphs