Another place in ScanOp::inferShapes needs refineShape=false (#1769)

Signed-off-by: Tung D. Le <[email protected]>
onnx · Oct 7, 2022 · f22d379 · f22d379
1 parent 38715dc
commit f22d379
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/Dialect/ONNX/ONNXOps.cpp b/src/Dialect/ONNX/ONNXOps.cpp
@@ -5111,7 +5111,8 @@ LogicalResult ONNXScanOp::inferShapes(
         scan_inputs().front().getType().cast<ShapedType>().getDimSize(0);
     unsqueezedShape.insert(unsqueezedShape.begin(), scanExtent);
     updateType(std::get<0>(vScanOutputValToTy), unsqueezedShape,
-        rankedScanTy.getElementType());
+        rankedScanTy.getElementType(), /*encoding=*/nullptr,
+        /*refineShape=*/false);
   }
 
   return success();