Skip to content

Commit

Permalink
[TF][TEST] add test_forward_reduce_any back (#4301)
Browse files Browse the repository at this point in the history
the test case was removed in #4181 for some reason
@tqchen @soiferj @zhiics
  • Loading branch information
yongwww authored and zhiics committed Nov 11, 2019
1 parent 18f8581 commit aeb5f13
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions tests/python/frontend/tensorflow/test_forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -2444,8 +2444,7 @@ def check_size(ishape):

#######################################################################
# All, Any, Max, Min
# -------------

# ------------------

def test_forward_reduce_all():
"""Test the All operator."""
Expand All @@ -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()
Expand Down

0 comments on commit aeb5f13

Please sign in to comment.