InferShapeFuncRegister Constructor and Destructor

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.

1
2
3

InferShapeFuncRegister (const std::string &operator_type, const InferShapeFunc &infer_shape_func)
InferShapeFuncRegister(const char *const operator_type, const InferShapeFunc &infer_shape_func)
~ InferShapeFuncRegister()

Parameters

Returns

Object of class InferShapeFuncRegister.

Constraints

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