diff options
Diffstat (limited to 'ethosu/vela/high_level_command_stream.py')
-rw-r--r-- | ethosu/vela/high_level_command_stream.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/ethosu/vela/high_level_command_stream.py b/ethosu/vela/high_level_command_stream.py index 9b0298bb..f9048a1e 100644 --- a/ethosu/vela/high_level_command_stream.py +++ b/ethosu/vela/high_level_command_stream.py @@ -34,11 +34,17 @@ class Box: assert self.start_coord[i] <= self.end_coord[i] def transform_with_strides_and_skirt( - self, strides, skirt, ifm_shape, npu_block_type, concat_axis=0, concat_offset=0, split_offset=None, k_height=1 + self, strides, skirt, ifm_shape, npu_block_type, concat_axis=0, concat_offset=0, split_offset=None, k_height=1, + upscaling_factor=1 ): new_start_coord = list(self.start_coord) new_end_coord = list(self.end_coord) + # Adjust for upscaling + if len(new_start_coord) == len(new_end_coord) == 4: + new_start_coord[1] = new_start_coord[1] // upscaling_factor + new_end_coord[1] = new_end_coord[1] // upscaling_factor + new_start_coord[concat_axis] -= concat_offset new_end_coord[concat_axis] -= concat_offset |