InferShapeRange

函数功能

注册算子的InferShapeRange函数。

用户需要为算子编写一个InferShapeRangeKernelFunc类型的函数，并使用该接口进行注册。

InferShapeRangeKernelFunc类型定义如下：

using InferShapeRangeKernelFunc = UINT32 (*)(InferShapeRangeContext *);

函数原型

OpImplRegisterV2 &InferShapeRange(InferShapeRangeKernelFunc infer_shape_range_func);

参数说明

返回值说明

返回算子的OpImplRegisterV2对象，该对象新增注册了InferShapeRange函数infer_shape_range_func。

约束说明

无。