SetInferShape

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

      
           OpDef &SetInferShape(gert::OpImplRegisterV2::InferShapeKernelFunc func)

Parameter

Input/Output

Description

func

Input

Shape inference function. The InferShapeKernelFunc type is defined as follows. For details about the input parameter type, see InferShapeContext.

           
                using InferShapeKernelFunc = UINT32 (*)(InferShapeContext *);

OpDef operator definition. For details, see OpDef.

None

Parent topic: OpDef