commit 68a04b129c8c341d4c1a621c04cfe5306390300c
author Jacob Bohlin <jacob.bohlin@arm.com> Mon Jul 13 11:39:36 2020 +0200
committer Tim Hall <tim.hall@arm.com> Tue Jul 14 15:31:21 2020 +0100
tree b1b097a38a3ab8181b3b6660fe82763ae3b0775a
parent 9b64ba0f5b45359db4c12fa77fc2caffb09f4432

MLBEDSW-2641: Fix crash for network with only CPU ops

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

ethosu/vela/tflite_writer.py

diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py
index cf40b5b..3f3b7b1 100644
--- a/ethosu/vela/tflite_writer.py
+++ b/ethosu/vela/tflite_writer.py
@@ -316,6 +316,7 @@
 
         scratch_tensors = [tens for tens in all_tensors if tens.name.endswith("scratch")]
 
+        scratch_fast_tensor = None
         for tens in all_tensors:
             if tens.name.endswith("scratch_fast"):
                 scratch_fast_tensor = tens