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.
Parent topic: InferShapeFuncRegister