Skip to content

Commit c76ced3

Browse files
committed
improve relu6 test
1 parent a88e5b6 commit c76ced3

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/test_backend.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -900,10 +900,9 @@ def test_tanh(self):
900900
_ = tf.identity(x_, name=_TFOUTPUT)
901901
self._run_test_case([_OUTPUT], {_INPUT: x_val}, rtol=1e-05)
902902

903-
@check_onnxruntime_incompatibility("Max")
904903
def test_relu6(self):
905-
x_val = np.array([0.5, 1.0, -0.5, -1.0], dtype=np.float32).reshape((2, 2))
906-
x = tf.placeholder(tf.float32, [2, 2], name=_TFINPUT)
904+
x_val = np.array([0.5, 1.0, -0.5, -1.0, 6, 7], dtype=np.float32).reshape((2, 3))
905+
x = tf.placeholder(tf.float32, x_val.shape, name=_TFINPUT)
907906
x_ = tf.nn.relu6(x)
908907
_ = tf.identity(x_, name=_TFOUTPUT)
909908
self._run_test_case([_OUTPUT], {_INPUT: x_val})

0 commit comments

Comments
 (0)