Skip to content

Commit bc492ac

Browse files
alnah005suhail
andauthored
fixed tuple error (#10216)
Co-authored-by: suhail <[email protected]>
1 parent 7766ab2 commit bc492ac

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

python/tvm/relay/frontend/tflite.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1995,9 +1995,8 @@ def convert_fully_connected(self, op):
19951995
# Change the output shape calculation based on keep_dim option
19961996
if keep_num_dims:
19971997
input_shape = _infer_shape(self.get_tensor_expr(input_tensor))
1998-
output_shape = list(input_shape)
1999-
output_shape[-1] = weight_tensor_shape[0]
2000-
out = _op.reshape(out, tuple(output_shape))
1998+
output_shape = input_shape[:-1] + tuple([weight_tensor_shape[0]])
1999+
out = _op.reshape(out, output_shape)
20012000

20022001
return out
20032002

0 commit comments

Comments
 (0)