From a5e8c1c3470409566723919e878b17297a52c54b Mon Sep 17 00:00:00 2001 From: Jacob Bohlin Date: Mon, 14 Jun 2021 13:33:39 +0200 Subject: MLBEDSW-4635: yolo_v3 output diff Fixed an issue where the scheduler would set the incorrect tensor layout. Signed-off-by: Jacob Bohlin Change-Id: I28abdf3f3c523d7da0cf8840316ece37dad364ab --- ethosu/vela/scheduler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu') 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) -- cgit v1.2.1