[Frontend][TFLite] Use axis.size instead of len(axis) (#8060)

The variable axis is an ndarray.
apache · May 18, 2021 · 94f009e · 94f009e
1 parent 0192750
commit 94f009e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py
@@ -3148,7 +3148,7 @@ def convert_expand_dims(self, op):
         input_expr = self.get_tensor_expr(input_tensors[0])
         axis = self.get_tensor_value(input_tensors[1])
         if isinstance(axis, np.ndarray):
-            assert len(axis) == 1, "only one value is expected."
+            assert axis.size == 1, "only one value is expected."
             axis = int(axis)
 
         ndims = len(input_tensors[0].tensor.ShapeAsNumpy())