From 04329626210a58516fb5126d7a99d6b73455c226 Mon Sep 17 00:00:00 2001 From: Krzysztof Parzyszek Date: Mon, 17 May 2021 14:40:21 -0500 Subject: [PATCH] [Frontend][TFLite] Use axis.size instead of len(axis) 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())