[ONNX] Reduce Onnx.Flatten op version (llvm#3560)

Signed-Off By: Vivek Khandelwal <[email protected]>
dbabokin · Jul 24, 2024 · 15cf710 · 15cf710
1 parent 003b06d
commit 15cf710
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/Conversion/TorchOnnxToTorch/DefaultDomainAtoF.cpp b/lib/Conversion/TorchOnnxToTorch/DefaultDomainAtoF.cpp
@@ -2535,7 +2535,7 @@ void mlir::torch::onnx_c::populateDefaultDomainAtoF(
         return success();
       });
   patterns.onOp(
-      "Flatten", 13, [](OpBinder binder, ConversionPatternRewriter &rewriter) {
+      "Flatten", 11, [](OpBinder binder, ConversionPatternRewriter &rewriter) {
         // Flatten means to partition the input tensor's dimensions
         // into a "left range" spanning 0 to axis - 1 and a "right range"
         // spanning axis to rank - 1.  Each range is then collapsed