diff options
author | Tim Hall <tim.hall@arm.com> | 2022-10-18 19:14:04 +0100 |
---|---|---|
committer | tim.hall <tim.hall@arm.com> | 2022-10-19 10:12:19 +0000 |
commit | d3d81b3ce138a48c0cddad7eb12710e26dad653e (patch) | |
tree | af370d21b457fcb4e75eb7af4faa55744be7ae4a | |
parent | b8b3e942006ea941cd52b719a1b59852803c1cd7 (diff) | |
download | ethos-u-vela-d3d81b3ce138a48c0cddad7eb12710e26dad653e.tar.gz |
MLBEDSW-7020: TRANSPOSE_CONV stride documentation is confusing
- The op contained supported operator checks for both the stride being
in the range 1 to 3, and being equal to 2. Whilst both are correct, only
the later is needed
- Removed the stride in the range 1 to 3 check for TRANSPOSE_CONV
- Regenerated the documentation
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I9789cdbd3ed65ce310f1529036abbac62296d2ca
-rw-r--r-- | SUPPORTED_OPS.md | 3 | ||||
-rw-r--r-- | ethosu/vela/tflite_supported_operators.py | 9 |
2 files changed, 7 insertions, 5 deletions
diff --git a/SUPPORTED_OPS.md b/SUPPORTED_OPS.md index bf7c2ba6..2efd017c 100644 --- a/SUPPORTED_OPS.md +++ b/SUPPORTED_OPS.md @@ -1,7 +1,7 @@ # Supported Ops This file was automatically generated by Vela using the `--supported-ops-report` parameter. -Vela version: `3.6.0rc1.dev23+g1f2496e.d20221017` +Vela version: `3.6.0rc1.dev18+g6ef0230.d20221018` This file complies with [**Gitiles Markdown syntax**](https://github.com/google/gitiles/blob/master/Documentation/markdown.md) @@ -330,7 +330,6 @@ This is a list of constraints that the TRANSPOSE_CONV operator must satisfy in o - Stride values for both width and height must be integer types - Dilation factor values for both width and height must be integer types -- Stride values for both width and height must be in the range [1, 3] - Dilation factor values for both width and height must be in the range [1, 2] - Dilated kernel height must be in the range [1, 64] - Product of dilated kernel width and height must be in the range [1, 4096] diff --git a/ethosu/vela/tflite_supported_operators.py b/ethosu/vela/tflite_supported_operators.py index 23bbac24..b8fe4b6a 100644 --- a/ethosu/vela/tflite_supported_operators.py +++ b/ethosu/vela/tflite_supported_operators.py @@ -235,14 +235,17 @@ class TFLiteSupportedOperators: self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_weights_limit) self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_bias_type) self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_bias_40bit) - # Depthwise Conv specific checks: - for op_type in TFLiteSupportedOperators.depthwise_convolution_ops: - self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_depth_multiplier) + # Remove stride contraint from Transpose Conv because it has a specific one (see below) + for op_type in TFLiteSupportedOperators.transpose_convolution_ops: + self.specific_constraints[op_type].remove(TFLiteSupportedOperators.constraint_stride_range) # Transpose Conv specific checks: for op_type in TFLiteSupportedOperators.transpose_convolution_ops: self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_tconv_stride) self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_tconv_same) self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_tconv_valid) + # Depthwise Conv specific checks: + for op_type in TFLiteSupportedOperators.depthwise_convolution_ops: + self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_depth_multiplier) # Pooling checks: for op_type in TFLiteSupportedOperators.pooling_ops: |