InferShapeFuncRegister Constructor and Destructor

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_factory.h>

Function Usage

Constructs and destructs an object of class InferShapeFuncRegister.

Prototype

APIs taking string arguments will be deprecated in future releases. Use APIs taking non-string arguments 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

Object of class InferShapeFuncRegister.

Restrictions

This API registers the operator InferShape function. It is referenced by other header files and is not directly called by operator developers.

Parent topic: InferShapeFuncRegister