diff options
author | Jacob Bohlin <jacob.bohlin@arm.com> | 2021-06-14 13:33:39 +0200 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-06-16 07:50:58 +0000 |
commit | a5e8c1c3470409566723919e878b17297a52c54b (patch) | |
tree | 3731d65935b59cc9553091fcc68326d7957c76f0 | |
parent | 4c00abb83487b3c8445f6db9140de44dec71efda (diff) | |
download | ethos-u-vela-a5e8c1c3470409566723919e878b17297a52c54b.tar.gz |
MLBEDSW-4635: yolo_v3 output diff
Fixed an issue where the scheduler would set the incorrect tensor
layout.
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I28abdf3f3c523d7da0cf8840316ece37dad364ab
-rw-r--r-- | ethosu/vela/scheduler.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py index 71007a32..dfb8867e 100644 --- a/ethosu/vela/scheduler.py +++ b/ethosu/vela/scheduler.py @@ -351,7 +351,7 @@ class Scheduler: if ps.primary_op: # Set tensor format to NHCWB16 for output FeatureMaps, if possible for output in ps.outputs: - if output.purpose != TensorPurpose.FeatureMap: + if output in self.sg.output_tensors or output.purpose != TensorPurpose.FeatureMap: continue if not output.needs_linear_format: output.set_format(TensorFormat.NHCWB16, arch) |