Do not create tosa.reshape for more than one dynamic dims.

Author: sahas3

lib/Conversion/TorchToTosa/TorchToTosa.cpp

@@ -1293,6 +1293,12 @@ class ConvertAtenSqueezeOp : public OpConversionPattern<AtenOpT> {
     auto newOutputTy = RankedTensorType::get(
         makeShapeLLVMCompatible(newOutputShape), resultElemTy);
 
+    if (newOutputTy.getNumDynamicDims() > 1) {
+      return rewriter.notifyMatchFailure(
+          op, "Squeeze op resulting in multiple dynamic dimensions in the "
+              "output is not supported.");
+    }
+
     auto reshapeOp = rewriter.create<tosa::ReshapeOp>(
         op->getLoc(),
         OpConversionPattern<AtenOpT>::getTypeConverter()->convertType(
@@ -3230,6 +3236,12 @@ LogicalResult ConvertAtenOp<AtenUnsqueezeOp>::matchAndRewrite(
         op, "Only tensor types are currently supported");
   }
 
+  if (selfType.getNumDynamicDims() > 1) {
+    return rewriter.notifyMatchFailure(
+        op,
+        "AtenUnsqueezeOp with multiple dynamic dimensions is not supported.");
+  }
+
   auto selfRank = selfType.getRank();
   auto selfElemTy = selfType.getElementType();
   if (!selfElemTy.isIntOrFloat()) {
@@ -3925,6 +3937,11 @@ class ConvertAtenMinMaxDimOp : public OpConversionPattern<AtenOpT> {
     }
 
     if (argMaxOp.getType() != indicesType) {
+      if (indicesType.getNumDynamicDims() > 1) {
+        return rewriter.notifyMatchFailure(
+            op, "AtenMaxDimOp resulting in multiple dynamic dimensions in the "
+                "output is not supported.");
+      }
       argMaxOp = rewriter.create<tosa::ReshapeOp>(
           op->getLoc(), indicesType, argMaxOp,
           tosa::getTosaConstShape(rewriter, op->getLoc(), reducedShape));