INFER_FUNC_REG

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

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.

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

Constraints

None

Parent topic: Data Structures and APIs