diff options
| -rw-r--r-- | ethosu/vela/tflite_reader.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py index be47cb13..9346b760 100644 --- a/ethosu/vela/tflite_reader.py +++ b/ethosu/vela/tflite_reader.py @@ -157,9 +157,11 @@ class TFLiteSubgraph: if op_type.startswith("DepthwiseConv2d") or op_type.startswith("Conv2D"): inputs[1] = clone_and_reshape_tensor(inputs[1], (1, 2, 3, 0)) + inputs[2] = clone_and_reshape_tensor(inputs[2], (0,)) if op_type.startswith("FullyConnected"): inputs[1] = clone_and_reshape_tensor(inputs[1], (1, 0)) + inputs[2] = clone_and_reshape_tensor(inputs[2], (0,)) if opt_serializer is not None: op.attrs = opt_serializer.deserialize(op_data) |