diff options
author | Rickard Bolin <rickard.bolin@arm.com> | 2022-06-09 13:07:17 +0000 |
---|---|---|
committer | Rickard Bolin <rickard.bolin@arm.com> | 2022-09-12 08:06:00 +0000 |
commit | 9ae34556663d09cc3dff19e53e68b57c8c940565 (patch) | |
tree | 51088224df14f66bd02afec67b933bd2ab192efd /ethosu/vela/cascade_builder.py | |
parent | 059166304f9ef47f0b916c1325700ed826f25581 (diff) | |
download | ethos-u-vela-9ae34556663d09cc3dff19e53e68b57c8c940565.tar.gz |
MLBEDSW-6613: Implement tile padding
Implement new padding mode which pads two edges of the IFM with the
current values of those edges
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: I8523e0cabdac80b48710703859003e33050cc150
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 94c856f4..3c105374 100644 --- a/ethosu/vela/cascade_builder.py +++ b/ethosu/vela/cascade_builder.py @@ -21,6 +21,7 @@ from collections import namedtuple from .numeric_util import round_up from .operation import NpuBlockType from .operation import Op +from .operation import Padding from .shape4d import Shape4D non_cascadable_blocks = ( @@ -99,6 +100,7 @@ class CascadeBuilder: and sched_op.parent_op.read_offsets[1] is None and self.element_wise_cascading_conformity(sched_op) and not sched_op.parent_op.type.is_resize_op() + and sched_op.parent_op.attrs.get("padding", None) != Padding.TILE ) def _is_mergeable(self, sched_op) -> bool: |