InferShapeRange
Description
Registers the InferShapeRange function of an operator.
You need to write a function of the InferShapeRangeKernelFunc type for the operator and use this API to register the function.
The InferShapeRangeKernelFunc type is defined as follows:
using InferShapeRangeKernelFunc = UINT32 (*)(InferShapeRangeContext *);
Prototype
OpImplRegisterV2 &InferShapeRange(InferShapeRangeKernelFunc infer_shape_range_func)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
infer_shape_range_func |
Input |
User-defined infer_shape_range_func function to be registered, which is of the InferShapeRangeKernelFunc type. |
Returns
OpImplRegisterV2 object of the operator, where the InferShapeRange function infer_shape_range_func is registered.
Restrictions
None
Parent topic: OpImplRegisterV2