Testcase check added to run in tf version above 1.15.0

apache · Feb 24, 2020 · a4d9023 · a4d9023
1 parent e32831f
commit a4d9023
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py
@@ -1132,7 +1132,8 @@ def test_all_reduce():
     _test_forward_reduce_quantized(_test_reduce_mean)
     _test_forward_reduce(_test_reduce_prod)
     _test_forward_reduce(_test_reduce_sum)
-    _test_forward_reduce(_test_reduce_any, dtype="bool")
+    if package_version.parse(tf.VERSION) >= package_version.parse('1.15.0'):
+        _test_forward_reduce(_test_reduce_any, dtype="bool")
 
 
 #######################################################################