INFER_FUNC_REG

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

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

Restrictions

None

Parent topic: Basic Data Structures and APIs of Graphs