diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py
index be47cb1..9346b76 100644
--- a/ethosu/vela/tflite_reader.py
+++ b/ethosu/vela/tflite_reader.py
@@ -157,9 +157,11 @@
 
         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)