SetInferShapeRange

Function

Registers the ShapeRange inference function. This API is required when the graph mode is used.

Prototype

      
           OpDef &SetInferShapeRange(gert::OpImplRegisterV2::InferShapeRangeKernelFunc func)

Parameters

Parameter

Input/Output

Description

func

Input

ShapeRange inference function. The InferShapeRangeKernelFunc type is defined as follows. For details about the input parameter type, see InferShapeRangeContext.

           
                using InferShapeRangeKernelFunc = UINT32 (*)(InferShapeRangeContext *);

Returns

OpDef operator definition. For details, see OpDef.

Constraints

None

Parent topic: OpDef