diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py index 78f04289806b5..162cc36a89e56 100644 --- a/python/tvm/relay/frontend/tflite.py +++ b/python/tvm/relay/frontend/tflite.py @@ -762,8 +762,8 @@ def convert_transpose(self, op): # axis in_axis = tuple(self.get_tensor_value(input_tensors[1])) - if in_axis is None: - out = _op.transpose(data=in_expr) + if not in_axis: + out = _op.transpose(in_expr) else: out = _op.transpose(in_expr, in_axis)