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/tensor.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/tensor.py')
| -rw-r--r-- | ethosu/vela/tensor.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index e9815845..65473b8d 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -600,7 +600,7 @@ class Tensor: 1, 1, 1, - [self.address_for_coordinate(start_coord, op_shape4D=op_shape4D), None, None, None], + [self.address_for_coordinate(start_coord, op_shape4D=op_shape4D), 0, 0, 0], ) if self.is_standard_fm: @@ -617,7 +617,7 @@ class Tensor: box_height0 = crossing_y - start_coord[1] box_width = crossing_x - start_coord[2] - addresses: List = [None] * 4 + addresses: List = [0] * 4 addresses[0] = self.address_for_coordinate(start_coord, op_shape4D=op_shape4D) if end_coord[2] > crossing_x: |