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():