Commit f5e2c44e by Ina Dobreva Committed by Tianqi Chen

[Bugfix][Frontend][TFlite] Fix wrong function call in TANH tests (#4517)

* Replace sigmoid() with tanh() in tests for TANH
parent f10944c9
......@@ -1030,7 +1030,7 @@ def _test_tanh(data):
""" One iteration of TANH """
with tf.Graph().as_default():
in_data = array_ops.placeholder(shape=data.shape, dtype=data.dtype)
out = math_ops.sigmoid(in_data)
out = math_ops.tanh(in_data)
compare_tflite_with_tvm(data, 'Placeholder:0', [in_data], [out])
def test_forward_tanh():
......
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