diff options
| author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-07-13 11:39:36 +0200 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2020-07-14 15:31:21 +0100 |
| commit | 68a04b129c8c341d4c1a621c04cfe5306390300c (patch) | |
| tree | b1b097a38a3ab8181b3b6660fe82763ae3b0775a /ethosu/vela/tflite_writer.py | |
| parent | 9b64ba0f5b45359db4c12fa77fc2caffb09f4432 (diff) | |
| download | ethos-u-vela-68a04b129c8c341d4c1a621c04cfe5306390300c.tar.gz | |
MLBEDSW-2641: Fix crash for network with only CPU ops
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: Ia9c70d62c6abc827cbdf73a8bb37afd595796741
Diffstat (limited to 'ethosu/vela/tflite_writer.py')
| -rw-r--r-- | ethosu/vela/tflite_writer.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_writer.py b/ethosu/vela/tflite_writer.py index cf40b5b5..3f3b7b1b 100644 --- a/ethosu/vela/tflite_writer.py +++ b/ethosu/vela/tflite_writer.py @@ -316,6 +316,7 @@ class TFLiteSerialiser: 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 |