MLBEDSW-7060: Bias tensor should be in 1D shape
Always make sure the bias is a 1D tensor.
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Change-Id: Ic0cb85d4fb9d2e07b4d1b7ac6059bffa432e28a3
diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py
index fa90ad9..9fe9ff5 100644
--- a/ethosu/vela/tflite_reader.py
+++ b/ethosu/vela/tflite_reader.py
@@ -141,8 +141,8 @@
inputs.append(None)
if inputs[-1] and inputs[-1].values is not None:
# Since bias tensor is used for both bias and scale,
- # a clone with a unique equivalence_id is needed
- inputs[-1] = clone_and_reshape_tensor(inputs[-1], (0,), True)
+ # a clone with a unique equivalence_id is needed.
+ inputs[-1] = clone_and_reshape_tensor(inputs[-1], None, True)
if opt_serializer is not None:
op.attrs = opt_serializer.deserialize(op_data)