From afb56ae11294bdf50e190cfafd76e507d5e2787d Mon Sep 17 00:00:00 2001 From: Johan Alfven Date: Fri, 27 Oct 2023 13:08:21 +0200 Subject: 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 --- SUPPORTED_OPS.md | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'SUPPORTED_OPS.md') diff --git a/SUPPORTED_OPS.md b/SUPPORTED_OPS.md index b22c5391..81704e53 100644 --- a/SUPPORTED_OPS.md +++ b/SUPPORTED_OPS.md @@ -19,7 +19,7 @@ limitations under the License. # Supported Ops This file was automatically generated by Vela using the `--supported-ops-report` parameter. -Vela version: `3.9.1.dev14+g7e03323` +Vela version: `3.9.1.dev16+gd230ce9.d20231030` This file complies with [**Gitiles Markdown syntax**](https://gerrit.googlesource.com/gitiles/+/HEAD/Documentation/markdown.md) @@ -159,9 +159,11 @@ This is a list of constraints that the CONV_2D operator must satisfy in order to - IFM depth must be a whole multiple of the filter kernel depth - Number of filter kernels must be equally divisible by the number of convolution groups - Dilation factor values for both width and height must be integer types -- Stride width must be greater than or equal to 1. - For stride widths greater than 3, the post-optimization stride needs to be less than or equal to 3. - Stride height must be between 1 and 3. +- Strides must fulfil the following criteria: + - Stride h must be between 1 and 3 when ofm height is greater than 1 + - Stride w must be between 1 and 3 when ofm height is greater than 1 or + stride w must be divisible by 2 or 3 and ifm width must be divisible + by stride_w/2 or stride_w/3 - Dilated kernel height must be in the range [1, 64] - Product of dilated kernel width and height must be in the range [1, 4096] - Weight tensor must be 8-bit -- cgit v1.2.1