[TF][TEST] add test_forward_reduce_any back (#4301)
authorYong Wu <ywu118@alumni.jh.edu>
Mon, 11 Nov 2019 17:24:52 +0000 (09:24 -0800)
committerZhi <5145158+zhiics@users.noreply.github.com>
Mon, 11 Nov 2019 17:24:52 +0000 (09:24 -0800)
the test case was removed in #4181 for some reason
@tqchen @soiferj @zhiics

tests/python/frontend/tensorflow/test_forward.py

index ce1d326..b8c980b 100644 (file)
@@ -2444,8 +2444,7 @@ def test_forward_size():
 
 #######################################################################
 # All, Any, Max, Min
-# -------------
-
+# ------------------
 
 def test_forward_reduce_all():
     """Test the All operator."""
@@ -2455,7 +2454,14 @@ def test_forward_reduce_all():
     tf.reduce_all(in_data, name="all")
     compare_tf_with_tvm([np_data], ['in_data:0'], 'all:0')
 
-
+def test_forward_reduce_any():
+    """Test the Any operator."""
+    np_data = np.random.choice([True, False], size=(5, 7, 11))
+    tf.reset_default_graph()
+    in_data = tf.placeholder(tf.bool, (5, 7, 11), name="in_data")
+    tf.reduce_any(in_data, name="any")
+    compare_tf_with_tvm([np_data], ['in_data:0'], 'any:0')
+    
 def test_forward_reduce_max():
     def check_max(ishape, axis, keepdims, dtype):
         tf.reset_default_graph()