diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py index 532318b804da..ef89fe80b75e 100644 --- a/python/tvm/relay/frontend/tflite.py +++ b/python/tvm/relay/frontend/tflite.py @@ -1898,10 +1898,6 @@ def _convert_arg_min_max(self, relay_op, op): def convert_arg_min(self, op): """Convert TFLite ARG_MIN""" - if self.is_quantized(op): - raise tvm.error.OpNotImplemented( - "TFlite quantized ARG_MIN operator is not supported yet." - ) return self._convert_arg_min_max(_op.argmin, op) def convert_arg_max(self, op): diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py index f60166702454..7adcdbd64404 100644 --- a/tests/python/frontend/tflite/test_forward.py +++ b/tests/python/frontend/tflite/test_forward.py @@ -3261,9 +3261,9 @@ def test_forward_arg_min_max(): """Arg min max""" # test quantized for data in [np.array(np.random.uniform(-100, 100, (3, 4)), dtype=np.uint8)]: - # There is no quantized version of ArgMin for axis in [None, 0, 1, -1]: _test_arg_min_max(math_ops.argmax, data, axis, True) + _test_arg_min_max(math_ops.argmin, data, axis, True) for data in [np.array(np.random.uniform(-100, 100, (3, 4)), dtype=np.float32)]: for axis in [None, 0, 1, -1]: