MLBEDSW-3352 Avoid ifm streaming for some cases

Changed so it is not allowed to do ifm-streaming for
TransposeConv and ResizeBilinear

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I85da279fae6202830c46e4a5500fb1b0dd6ef542

1 files changed, 7 insertions, 1 deletions

diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py
index d6e890a..943c590 100644
--- a/ethosu/vela/scheduler.py
+++ b/ethosu/vela/scheduler.py
@@ -557,6 +557,12 @@ class DynamicProgrammingScheduler:
 
         return strat_data
 
+    def avoid_ifm_streaming(self, ps):
+        for op in ps.ops:
+            if op.type in (Op.Conv2DBackpropInputSwitchedBias, Op.ResizeBilinear):
+                return True
+        return False
+
     @lru_cache(maxsize=None)
     def search_output(self, ps):
 
@@ -565,7 +571,7 @@ class DynamicProgrammingScheduler:
 
         candidate_list.extend(self.search_weight_streaming_output(ps))
 
-        if self.options.use_ifm_streaming:
+        if self.options.use_ifm_streaming and not self.avoid_ifm_streaming(ps):
             candidate_list.extend(self.search_ifm_streaming_output(ps))
 
         best = self.filter_pareto_frontier(candidate_list, remove_equally_good_candidates=True)