diff options
| -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) |