Skip to content

Commit

Permalink
[TF][TEST] add test_forward_reduce_any back
Browse files Browse the repository at this point in the history
the test case was removed in apache#4181 for some reason
@tqchen @soiferj @zhiics
  • Loading branch information
yongwww authored Nov 11, 2019
1 parent 1d24366 commit e927da5
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 @@ -2432,8 +2432,7 @@ def check_size(ishape):

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

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

def test_forward_reduce_all():
"""Test the All operator."""
Expand All @@ -2443,7 +2442,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 e927da5

Please sign in to comment.