fixed tuple error (#10216)

Co-authored-by: suhail <suhail@expedera.com>
apache · May 27, 2022 · bc492ac · bc492ac
1 parent 7766ab2
commit bc492ac
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py
@@ -1995,9 +1995,8 @@ def convert_fully_connected(self, op):
         # Change the output shape calculation based on keep_dim option
         if keep_num_dims:
             input_shape = _infer_shape(self.get_tensor_expr(input_tensor))
-            output_shape = list(input_shape)
-            output_shape[-1] = weight_tensor_shape[0]
-            out = _op.reshape(out, tuple(output_shape))
+            output_shape = input_shape[:-1] + tuple([weight_tensor_shape[0]])
+            out = _op.reshape(out, output_shape)
 
         return out