diff options
author | Johan Alfven <johan.alfven@arm.com> | 2023-10-27 13:08:21 +0200 |
---|---|---|
committer | Johan Alfven <johan.alfven@arm.com> | 2023-10-31 10:18:33 +0100 |
commit | afb56ae11294bdf50e190cfafd76e507d5e2787d (patch) | |
tree | bf377821860dbba9f92bad3395d29b2560c840ef /ethosu/vela/utils.py | |
parent | 909923a2461087f64a40e07be7752e8923f70ae7 (diff) | |
download | ethos-u-vela-afb56ae11294bdf50e190cfafd76e507d5e2787d.tar.gz |
MLBEDSW-8201: [MLCE] Extended stride support for CONV_2D
- Added support for stride_h > 3 when ofm height is 1
- Added support for stride_w > 3 when ofm width is 1
- Updated constraints
- Updated tests
- Updated SUPPORTED_OPS.md
Change-Id: I8f89909b05a0f052df5f03702966cee50da61cfc
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Diffstat (limited to 'ethosu/vela/utils.py')
-rw-r--r-- | ethosu/vela/utils.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/utils.py b/ethosu/vela/utils.py index 11c253c0..ee501f3f 100644 --- a/ethosu/vela/utils.py +++ b/ethosu/vela/utils.py @@ -96,7 +96,7 @@ def calc_resize_factor(ifm_width: int, stride_x: int) -> tuple[int, int]: # In case it is not divisible, check if the resize factor is # divisible by any of the hw_supported_strides. If it is, re-compute # the resize factor to be the value that leads us to - # reach a hw supported stride. The IFM width needs to be divisible by the new stride. + # reach a hw supported stride. The IFM width needs to be divisible by the new resize factor. # E.g.: IFM width = 133, stride = 14, filter width = 7 can be # optimised to IFM width = 19, stride = 2, filter width = 7 using # a resize factor of 7. The final stride is 2 which is |