Skip to content

T
tic
Unverified Commit 7226c010 by Tianqi Chen Committed by GitHub
Options

[TEST] Hotfix CI outrage after TF in docker update (#2781)

parent 92a00489
Showing with 2 additions and 2 deletions
nnvm/tests/python/frontend/tensorflow/test_forward.py
...@@ -1125,7 +1125,7 @@ def test_forward_leaky_relu(): ...@@ -1125,7 +1125,7 @@ def test_forward_leaky_relu():
with tf.Graph().as_default(): with tf.Graph().as_default():
in1 = tf.placeholder(shape=inp_array.shape, dtype=inp_array.dtype) in1 = tf.placeholder(shape=inp_array.shape, dtype=inp_array.dtype)
tf.nn.leaky_relu(in1, alpha=0.4) tf.nn.leaky_relu(in1, alpha=0.4)
compare_tf_with_tvm(inp_array, 'Placeholder:0', 'LeakyRelu/mul:0') compare_tf_with_tvm(inp_array, 'Placeholder:0', 'LeakyRelu:0')
def test_forward_elu(): def test_forward_elu():
ishape = (1, 3, 10, 10) ishape = (1, 3, 10, 10)
......
tests/python/frontend/tensorflow/test_forward.py
...@@ -1033,7 +1033,7 @@ def test_forward_leaky_relu(): ...@@ -1033,7 +1033,7 @@ def test_forward_leaky_relu():
with tf.Graph().as_default(): with tf.Graph().as_default():
in1 = tf.placeholder(shape=inp_array.shape, dtype=inp_array.dtype) in1 = tf.placeholder(shape=inp_array.shape, dtype=inp_array.dtype)
tf.nn.leaky_relu(in1, alpha=0.4) tf.nn.leaky_relu(in1, alpha=0.4)
compare_tf_with_tvm(inp_array, 'Placeholder:0', 'LeakyRelu/mul:0') compare_tf_with_tvm(inp_array, 'Placeholder:0', 'LeakyRelu:0')
def test_forward_elu(): def test_forward_elu():
ishape = (1, 3, 10, 10) ishape = (1, 3, 10, 10)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment