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 /ethosu/vela/tflite_supported_operators.py | |
| 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
Diffstat (limited to 'ethosu/vela/tflite_supported_operators.py')
| -rw-r--r-- | ethosu/vela/tflite_supported_operators.py | 9 |
1 files changed, 6 insertions, 3 deletions
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: |