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