From 8b5c06221a71969ee95e161be77a31680c57e2fc Mon Sep 17 00:00:00 2001 From: Krzysztof Parzyszek Date: Tue, 18 May 2021 17:30:00 -0500 Subject: [PATCH] [Frontend][TFLite] Use axis.size instead of len(axis) (#8060) The variable axis is an ndarray. --- python/tvm/relay/frontend/tflite.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py index 6b14a6f58e60..656eb597e587 100644 --- 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())