SetInferDataType

Function

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

Prototype

      
           OpDef &SetInferDataType(gert::OpImplRegisterV2::InferDataTypeKernelFunc func)

Parameters

Parameter

Input/Output

Description

func

Input

DataType inference function. The InferDataTypeKernelFunc type is defined as follows. For details about the input parameter type, see InferDataTypeContext.

           
                using InferDataTypeKernelFunc = UINT32 (*)(InferDataTypeContext *);

Returns

OpDef operator definition. For details, see OpDef.

Constraints

None

Parent topic: OpDef