aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tensor.py
diff options
context:
space:
mode:
authorTim Hall <tim.hall@arm.com>2020-07-28 17:40:50 +0100
committertim.hall <tim.hall@arm.com>2020-08-21 15:34:27 +0000
commite3786ac6bee55c41b69c0cfb94344f3372f8c991 (patch)
tree9c313936fd886a0996e5df9a5f25011b6f5f3936 /ethosu/vela/tensor.py
parent67e0d8f24fcb86115e834acd19dc57027b03ea4f (diff)
downloadethos-u-vela-e3786ac6bee55c41b69c0cfb94344f3372f8c991.tar.gz
MLBEDSW-2679: Tensor quant comparison is incorrect
- Fixed bug with the supported operator check rejecting operators based upon an incorrect comparison of the tensor quantisations Signed-off-by: Tim Hall <tim.hall@arm.com> Change-Id: Ibd0eb50077465d2c515c6ee10394d9b43cdf730c
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r--ethosu/vela/tensor.py19
1 files changed, 6 insertions, 13 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index 66bed59d..5fdea979 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -184,19 +184,6 @@ class QuantizationParameters:
__repr__ = __str__
- def __eq__(self, other):
- if other is None:
- return False
- if not isinstance(other, QuantizationParameters):
- return False
-
- pairs = ((getattr(self, s), getattr(other, s)) for s in QuantizationParameters.__slots__)
-
- return all(np.array_equal(a, b) for a, b in pairs)
-
- def __ne__(self, other):
- return not self == other
-
def clone(self):
res = QuantizationParameters()
res.min = self.min
@@ -232,6 +219,12 @@ class QuantizationParameters:
return res
+ def is_scaling_equal(self, other):
+ if other is None or not isinstance(other, QuantizationParameters):
+ return False
+
+ return self.scale_f32 == other.scale_f32 and self.zero_point == other.zero_point
+
def create_const_tensor(name, shape, dtype, values, value_dtype=None, purpose=TensorPurpose.Unknown, quantization=None):
# Tensor