From 71525176b7aee5ed1dc9f7ae09e11a1884238615 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Sat, 29 Aug 2020 15:09:57 +0100 Subject: MLBEDSW-2804: Bias data type check wrong - Fixed bias check to use quantised values. Signed-off-by: Tim Hall Change-Id: I6d87439938b9b5aeec87814e5a30d59fd06d5748 --- ethosu/vela/supported_operators.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'ethosu/vela') diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index 8ec77207..591f0801 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -421,8 +421,8 @@ class SupportedOperators: # check if values fits in 40-bit if bias_tensor.dtype == DataType.int64: - for value in bias_tensor.values: - if not (-(1 << 39) <= value < (1 << 39)): + for quant_value in bias_tensor.quant_values: + if not (-(1 << 39) <= quant_value < (1 << 39)): return False return True -- cgit v1.2.1