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