diff options
| -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: |