CI tests

Author: srkreddy1238

tests/python/relax/test_frontend_tflite.py

@@ -708,10 +708,10 @@ def main(x: R.Tensor((5, 30), dtype="float32")) -> R.Tensor(out_shape, dtype="in
     "data, kernel, data_format, strides, padding",
     [
         # Tf on CI (CPU) support only NHWC
-        # ((1, 128, 128, 32), (3, 3, 32, 32), "NHWC", (1, 1, 1, 1), "SAME"),
-        # ((1, 128, 128, 32), (3, 3, 32, 32), "NHWC", (1, 1, 1, 1), "VALID"),
-        ((1, 32, 128, 128), (3, 3, 32, 32), "NCHW", (1, 1, 1, 1), "SAME"),
-        ((1, 32, 128, 128), (3, 3, 32, 32), "NCHW", (1, 1, 1, 1), "VALID"),
+        ((1, 128, 128, 32), (3, 3, 32, 32), "NHWC", (1, 1, 1, 1), "SAME"),
+        ((1, 128, 128, 32), (3, 3, 32, 32), "NHWC", (1, 1, 1, 1), "VALID"),
+        # ((1, 32, 128, 128), (3, 3, 32, 32), "NCHW", (1, 1, 1, 1), "SAME"),
+        # ((1, 32, 128, 128), (3, 3, 32, 32), "NCHW", (1, 1, 1, 1), "VALID"),
     ],
 )
 def test_conv2d(data, kernel, data_format, strides, padding):
@@ -742,10 +742,10 @@ def func(self, data, kernel):
     "data, kernel, data_format, strides, padding",
     [
         # Tf on CI (CPU) support only NHWC
-        # ((1, 128, 128, 32), (2, 2), "NHWC", (1, 1, 1, 1), "SAME"),
-        # ((1, 128, 128, 32), (2, 2), "NHWC", (1, 1, 1, 1), "VALID"),
-        ((1, 32, 128, 128), (3, 3), "NCHW", (1, 1, 1, 1), "SAME"),
-        ((1, 32, 128, 128), (3, 3), "NCHW", (1, 1, 1, 1), "VALID"),
+        ((1, 128, 128, 32), (2, 2), "NHWC", (1, 1, 1, 1), "SAME"),
+        ((1, 128, 128, 32), (2, 2), "NHWC", (1, 1, 1, 1), "VALID"),
+        # ((1, 32, 128, 128), (3, 3), "NCHW", (1, 1, 1, 1), "SAME"),
+        # ((1, 32, 128, 128), (3, 3), "NCHW", (1, 1, 1, 1), "VALID"),
     ],
 )
 def test_pool_2d(pool, data, kernel, data_format, strides, padding):