From ce8134c09c8cf352fc9321d7504d4c79b6740428 Mon Sep 17 00:00:00 2001 From: Ina_Dobreva Date: Fri, 13 Dec 2019 10:53:26 +0000 Subject: [PATCH] [Bugfix][Frontend][TFlite] Fix wrong function call in TANH tests * Replace sigmoid() with tanh() in tests for TANH --- tests/python/frontend/tflite/test_forward.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py index 12ea429983e8..fd43ff3f4229 100644 --- a/tests/python/frontend/tflite/test_forward.py +++ b/tests/python/frontend/tflite/test_forward.py @@ -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():