commit a5e8c1c3470409566723919e878b17297a52c54b
author Jacob Bohlin <jacob.bohlin@arm.com> Mon Jun 14 13:33:39 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Wed Jun 16 07:50:58 2021 +0000
tree 3731d65935b59cc9553091fcc68326d7957c76f0
parent 4c00abb83487b3c8445f6db9140de44dec71efda

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

ethosu/vela/scheduler.py

diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py
index 71007a3..dfb8867 100644
--- a/ethosu/vela/scheduler.py
+++ b/ethosu/vela/scheduler.py
@@ -351,7 +351,7 @@
             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)