[Flaky] TFLite quantized conv test (apache#6084)

trevor-m · Sep 2, 2020 · 0f8e7b5 · 0f8e7b5
1 parent 6ceefe3
commit 0f8e7b5
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py
@@ -785,6 +785,7 @@ def representative_data_gen():
 def _test_tflite2_quantized_depthwise_convolution(input_shape, kernel_shape,
         dilations, strides, padding, data_format, depth_multiplier):
     """One iteration of TFLite2 quantized depthwise convolution with given shapes and attributes"""
+
     data_format = "channels_last" if "NHWC" else "channels_first"
     data = np.random.uniform(0, 1, input_shape).astype('float32')
     kernel = np.random.uniform(0, 1, kernel_shape).astype('float32')
@@ -914,13 +915,14 @@ def test_forward_convolution():
         _test_tflite2_quantized_convolution([1, 17, 17, 19], [3, 3, 19, 19], [1, 1], [2, 2], 'VALID', 'NHWC')
         _test_tflite2_quantized_convolution([1, 17, 17, 124], [1, 1, 124, 19], [1, 1], [1, 1], 'SAME', 'NHWC')
 
+        # Disable as tests are flaky - https://github.com/apache/incubator-tvm/issues/6064
         # depthwise convolution
-        _test_tflite2_quantized_depthwise_convolution([1, 8, 8, 128], [1, 1, 128, 1], [1, 1], [1, 1],
-                                                      'SAME', 'NHWC', 1)
-        _test_tflite2_quantized_depthwise_convolution([1, 17, 17, 12], [3, 3, 12, 1], [1, 1], [2, 2],
-                                                      'VALID', 'NHWC', 1)
-        _test_tflite2_quantized_depthwise_convolution([1, 24, 24, 3], [7, 7, 3, 8], [1, 1], [2, 2],
-                                                      'SAME', 'NHWC', 8)
+        # _test_tflite2_quantized_depthwise_convolution([1, 8, 8, 128], [1, 1, 128, 1], [1, 1], [1, 1],
+        #                                               'SAME', 'NHWC', 1)
+        # _test_tflite2_quantized_depthwise_convolution([1, 17, 17, 12], [3, 3, 12, 1], [1, 1], [2, 2],
+        #                                               'VALID', 'NHWC', 1)
+        # _test_tflite2_quantized_depthwise_convolution([1, 24, 24, 3], [7, 7, 3, 8], [1, 1], [2, 2],
+        #                                               'SAME', 'NHWC', 8)