diff --git a/tests/python/relax/test_frontend_tflite.py b/tests/python/relax/test_frontend_tflite.py index e7d81cf5fea0..fd3def052127 100644 --- a/tests/python/relax/test_frontend_tflite.py +++ b/tests/python/relax/test_frontend_tflite.py @@ -511,6 +511,7 @@ def main(x: R.Tensor((1, 30), dtype="float32")) -> R.Tensor((1, 30), dtype="floa (tf.math.cos, R.cos), (tf.math.sin, R.sin), (tf.math.exp, R.exp), + (tf.math.log, R.log), (tf.math.negative, R.negative), (tf.round, R.round), (tf.math.rsqrt, R.rsqrt), @@ -542,6 +543,8 @@ def main(x: R.Tensor((1, 30), dtype="float32")) -> R.Tensor((1, 30), dtype="floa [ (tf.math.less, R.less), (tf.math.less_equal, R.less_equal), + (tf.math.greater, R.greater), + (tf.math.greater_equal, R.greater_equal), (tf.math.equal, R.equal), (tf.math.not_equal, R.not_equal), ],