commit 258ebbaee96fb24543bde99c3e48ae69d9d5486f
author Jacob Bohlin <jacob.bohlin@arm.com> Mon Aug 31 10:44:35 2020 +0200
committer Jacob Bohlin <jacob.bohlin@arm.com> Mon Aug 31 14:21:57 2020 +0200
tree 0507d4e78417d50e942dba121a626a7a2556cb60
parent 749bfd5d2e28dbced80782fcc18cdcd2f9475555

MLBEDSW-2915: Added None check for bias before checking restrictions

Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I04618fd0d29075e7d3f8f27a320129603f045163

ethosu/vela/supported_operators.py

diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py
index 591f080..7cff0ee 100644
--- a/ethosu/vela/supported_operators.py
+++ b/ethosu/vela/supported_operators.py
@@ -416,11 +416,11 @@
 
     def check_bias_restrictions(self, bias_tensor):
         # check data type
-        if bias_tensor.dtype not in (DataType.int32, DataType.int64):
+        if bias_tensor is not None and bias_tensor.dtype not in (DataType.int32, DataType.int64):
             return False
 
         # check if values fits in 40-bit
-        if bias_tensor.dtype == DataType.int64:
+        if bias_tensor is not None and bias_tensor.dtype == DataType.int64:
             for quant_value in bias_tensor.quant_values:
                 if not (-(1 << 39) <= quant_value < (1 << 39)):
                     return False