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