SetInferShapeRange
功能说明
使用图模式时，需要调用该接口注册ShapeRange推导函数。
函数原型
Text
OpDef &SetInferShapeRange(gert::OpImplRegisterV2::InferShapeRangeKernelFunc func)
参数说明
参数 | 输入/输出 | 说明 |
|---|---|---|
func | 输入 | ShapeRange推导函数。InferShapeRangeKernelFunc类型定义如下： using InferShapeRangeKernelFunc = UINT32 (*)(InferShapeRangeContext *); |
返回值说明
OpDef算子定义，OpDef请参考OpDef。
约束说明
无