From d3d81b3ce138a48c0cddad7eb12710e26dad653e Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Tue, 18 Oct 2022 19:14:04 +0100 Subject: 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 Change-Id: I9789cdbd3ed65ce310f1529036abbac62296d2ca --- ethosu/vela/tflite_supported_operators.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'ethosu') 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: -- cgit v1.2.1