InferShapeFuncRegister Constructor and Destructor

Description

Constructs or destructs an object of class InferShapeFuncRegister.

Prototype

APIs using strings will be deprecated in later versions. Use APIs supporting non-string data instead.

InferShapeFuncRegister (const std::string &operator_type, const InferShapeFunc &infer_shape_func);

InferShapeFuncRegister(const char *const operator_type, const InferShapeFunc &infer_shape_func);

~ InferShapeFuncRegister()

Parameters

Parameter

Input/Output

Description

operator_type

Input

Operator type.

infer_shape_func

Input

Infershape function of the operator.

Returns

An object of the InferShapeFuncRegister class.

Restrictions

This API registers the operator infershape function and is referenced by other header files. Therefore, it is not directly called by the operator developer.