Overview
Shape inference API of custom operators. It is applicable to the scenario where the operator performs shape and data type inference based on the inference context.
Header Files to Be Included
1 | #include <graph/custom_op.h> |
Public Member Functions
virtual graphStatus InferShape(gert::InferShapeContext *ctx) = 0 virtual graphStatus InferDataType(gert::InferDataTypeContext *ctx) = 0
Parent topic: ShapeInferOp