diff options
Diffstat (limited to 'ethosu/vela/tflite_reader.py')
-rw-r--r-- | ethosu/vela/tflite_reader.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py index fa90ad9e..9fe9ff58 100644 --- a/ethosu/vela/tflite_reader.py +++ b/ethosu/vela/tflite_reader.py @@ -141,8 +141,8 @@ class TFLiteSubgraph: 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) |