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

The variable axis is an ndarray.
neo-ai · Jun 17, 2021 · 8b5c062 · 8b5c062
1 parent 2e4618a
commit 8b5c062
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())