diff options
author | Charles Xu <charles.xu@arm.com> | 2020-06-23 12:42:28 +0200 |
---|---|---|
committer | Charles Xu <charles.xu@arm.com> | 2020-07-02 09:38:12 +0200 |
commit | 04ce34c07cf7fcd0a8ddb29df6f59846862b24f2 (patch) | |
tree | f6e9e5aa5a6e01beb7058c9849f771fa52a75bff /ethosu/vela/tflite_writer.py | |
parent | 5f47c0511d3fc6b54aba331b58c85c3970b61718 (diff) | |
download | ethos-u-vela-04ce34c07cf7fcd0a8ddb29df6f59846862b24f2.tar.gz |
MLBEDSW-2340: Make the tensor address default None
Signed-off-by: Charles Xu <charles.xu@arm.com>
Change-Id: I53d9d56acee57cff208dccb4822c1f1a461c416d
Diffstat (limited to 'ethosu/vela/tflite_writer.py')
-rw-r--r-- | ethosu/vela/tflite_writer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py index 7e805e31..4aa23b5f 100644 --- a/ethosu/vela/tflite_writer.py +++ b/ethosu/vela/tflite_writer.py @@ -401,7 +401,7 @@ class TFLiteSerialiser: # 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) |