aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tensor.py
diff options
context:
space:
mode:
authorPatrik Gustavsson <patrik.gustavsson@arm.com>2020-11-03 10:24:08 +0100
committerPatrik Gustavsson <patrik.gustavsson@arm.com>2020-11-03 16:21:21 +0100
commit34359589655e62d0f2cadde53320de71add04e94 (patch)
treeaf2a10c420f9bedc1984225bd2f16717fad11af9 /ethosu/vela/tensor.py
parent7e1d1d181d5334bf50fc2e1f39f11c20a22fc341 (diff)
downloadethos-u-vela-34359589655e62d0f2cadde53320de71add04e94.tar.gz
MLBEDSW-2868 Separate scale+bias tensors
Separate scale+bias tensors by different equivilence_id. Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com> Change-Id: I674341950bc001ac6e4015206995f048a0dfee75
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r--ethosu/vela/tensor.py3
1 files changed, 3 insertions, 0 deletions
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 "<nng.Tensor '%s' shape=%s dtype=%s>" % (self.name, self.shape, self.dtype)