[ET-VK] fix index error bug in ViewCopyToSqueezeUnsqueezePass

See T214560872
#8226 added the pass to the partition preprocess pass list, so now it runs on all exports.
This uncovered a bug in the squeeze dims finding function in the mobilenet test case.

Differential Revision: [D69254910](https://our.internmc.facebook.com/intern/diff/D69254910/)

[ghstack-poisoned]

Author: Nathanael See

backends/transforms/view_copy_to_squeeze_unsqueeze.py

@@ -47,15 +47,18 @@ def find_squeeze_dims(
         j = 0
         idx = []
         while i < len(input_shape):
-            if input_shape[i] != view_shape[j]:
-                if input_shape[i] == 1:
-                    idx.append(i)
-                    j -= 1
-                    # continue to check remaining dims are equal
-                else:
-                    return None
-            i += 1
-            j += 1
+            if j < len(view_shape) and input_shape[i] == view_shape[j]:
+                i += 1
+                j += 1
+            elif input_shape[i] == 1:
+                # squeeze axis on i and check next dim
+                idx.append(i)
+                i += 1
+            else:
+                return None
+        # If there are remaining dimensions in view_shape, shapes do not match
+        if j < len(view_shape):
+            return None
         return idx
 
     def find_unsqueeze_dim(