Skip to content

T
tic
Commit f5e2c44e by Ina Dobreva Committed by Tianqi Chen
Options

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

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