SetInferDataType
Function Usage
Registers the DataType inference function.
Prototype
OpDef &SetInferDataType(gert::OpImplRegisterV2::InferDataTypeKernelFunc func);
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
func |
Input |
DataType inference function. The InferDataTypeKernelFunc type is defined as follows: using InferDataTypeKernelFunc = UINT32 (*)(InferDataTypeContext *); |
Returns
OpDef operator definition. For details, see OpDef.
Constraints
None
Parent topic: OpDef