Improve the if condition for empty tuple

apache · Aug 9, 2019 · 889e99d · 889e99d
1 parent 3776688
commit 889e99d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)