diff --git a/src/Dialect/ONNX/ONNXOps.cpp b/src/Dialect/ONNX/ONNXOps.cpp index c204958d4a..6a662fce8e 100644 --- a/src/Dialect/ONNX/ONNXOps.cpp +++ b/src/Dialect/ONNX/ONNXOps.cpp @@ -5111,7 +5111,8 @@ LogicalResult ONNXScanOp::inferShapes( scan_inputs().front().getType().cast().getDimSize(0); unsqueezedShape.insert(unsqueezedShape.begin(), scanExtent); updateType(std::get<0>(vScanOutputValToTy), unsqueezedShape, - rankedScanTy.getElementType()); + rankedScanTy.getElementType(), /*encoding=*/nullptr, + /*refineShape=*/false); } return success();