MLBEDSW-3352 Fix ifm end_coord for upsampling
-Fix for end_coord for upsampling
-Remove restriction for ifm streaming
-Added restriction for cascading on ResizeBilinear
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I384abf12cfe8ac9ce7b76066b709600ea901b248
diff --git a/ethosu/vela/scheduler.py b/ethosu/vela/scheduler.py
index 943c590..7347b5a 100644
--- a/ethosu/vela/scheduler.py
+++ b/ethosu/vela/scheduler.py
@@ -557,12 +557,6 @@
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):
@@ -571,7 +565,7 @@
candidate_list.extend(self.search_weight_streaming_output(ps))
- if self.options.use_ifm_streaming and not self.avoid_ifm_streaming(ps):
+ if self.options.use_ifm_streaming:
candidate_list.extend(self.search_ifm_streaming_output(ps))
best = self.filter_pareto_frontier(candidate_list, remove_equally_good_candidates=True)
@@ -657,6 +651,8 @@
if len(op.outputs) > 1 or len(op.outputs[0].consumer_list) > 1:
# The op has consumers in other subgraphs
return True
+ if op.type == Op.ResizeBilinear:
+ return True
return False
def search_ifm_streaming_partial(self, ps, block_config):