From 34359589655e62d0f2cadde53320de71add04e94 Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Tue, 3 Nov 2020 10:24:08 +0100 Subject: MLBEDSW-2868 Separate scale+bias tensors Separate scale+bias tensors by different equivilence_id. Signed-off-by: Patrik Gustavsson Change-Id: I674341950bc001ac6e4015206995f048a0dfee75 --- ethosu/vela/tensor.py | 3 +++ ethosu/vela/tflite_reader.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 0fcf9a97..8786d362 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -806,6 +806,9 @@ class Tensor: return True + def set_random_equivalence_id(self): + self.equivalence_id = uuid.uuid4() + def __str__(self): return "" % (self.name, self.shape, self.dtype) diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py index f4f48049..82feddd9 100644 --- a/ethosu/vela/tflite_reader.py +++ b/ethosu/vela/tflite_reader.py @@ -161,6 +161,9 @@ class TFLiteSubgraph: inputs.append(None) if inputs[-1] and inputs[-1].values is not None: inputs[-1] = clone_and_reshape_tensor(inputs[-1], (0,)) + # Since bias tensor is used for both bias and scale, + # set different equivalence_id for all bias tensors + inputs[-1].set_random_equivalence_id() if opt_serializer is not None: op.attrs = opt_serializer.deserialize(op_data) -- cgit v1.2.1