diff options
Diffstat (limited to 'ethosu/vela/cascade_builder.py')
| -rw-r--r-- | ethosu/vela/cascade_builder.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/cascade_builder.py b/ethosu/vela/cascade_builder.py index e4fa67e9..0b50eaa8 100644 --- a/ethosu/vela/cascade_builder.py +++ b/ethosu/vela/cascade_builder.py @@ -174,6 +174,8 @@ class CascadeBuilder: or current_op not in ref_cost or not self._is_cascadable(current_op, ref_cost[current_op]) or producer.ofm.shape != current_op.ifm.shape + or current_op.requires_full_ifm + or producer.requires_full_ofm ): # Current op has already been processed or cannot be cascaded break |