commit 04ce34c07cf7fcd0a8ddb29df6f59846862b24f2
author Charles Xu <charles.xu@arm.com> Tue Jun 23 12:42:28 2020 +0200
committer Charles Xu <charles.xu@arm.com> Thu Jul 02 09:38:12 2020 +0200
tree f6e9e5aa5a6e01beb7058c9849f771fa52a75bff
parent 5f47c0511d3fc6b54aba331b58c85c3970b61718

MLBEDSW-2340: Make the tensor address default None

Signed-off-by: Charles Xu <charles.xu@arm.com>
Change-Id: I53d9d56acee57cff208dccb4822c1f1a461c416d

ethosu/vela/tflite_writer.py

diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py
index 7e805e3..4aa23b5 100644
--- a/ethosu/vela/tflite_writer.py
+++ b/ethosu/vela/tflite_writer.py
@@ -401,7 +401,7 @@
         # Ensure that the order of the offsets match the order of the tensors
         for tens, idx in self.tensor_map.items():
             # Set offsets for tensor allocated in Tensor Arena or in the scratch_fast area
-            if tens.mem_type in set((MemType.Scratch, MemType.Scratch_fast)):
+            if tens.mem_type in set((MemType.Scratch, MemType.Scratch_fast)) and tens.address is not None:
                 offsets[idx] = np.int32(tens.address)
 
         metadata_buffer = np.array([version, subgraph_idx, nbr_tensors] + offsets)