Overview
Registers the operator InferShape function. This API is referenced by other header files and is not directly called by operator developers.
Header Files to Be Included
1 | #include <graph/operator_factory.h> |
Public Member Functions
InferShapeFuncRegister (const std::string &operator_type, const InferShapeFunc &infer_shape_func) InferShapeFuncRegister(const char *const operator_type, const InferShapeFunc &infer_shape_func) ~ InferShapeFuncRegister()
Parent topic: InferShapeFuncRegister