We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7766ab2 commit bc492acCopy full SHA for bc492ac
python/tvm/relay/frontend/tflite.py
@@ -1995,9 +1995,8 @@ def convert_fully_connected(self, op):
1995
# Change the output shape calculation based on keep_dim option
1996
if keep_num_dims:
1997
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))
+ output_shape = input_shape[:-1] + tuple([weight_tensor_shape[0]])
+ out = _op.reshape(out, output_shape)
2001
2002
return out
2003
0 commit comments