From f49b6e250165e5f0f41befb0d9984ddafe39c41e Mon Sep 17 00:00:00 2001 From: Johan Alfven Date: Wed, 15 Nov 2023 10:11:31 +0100 Subject: MLBEDSW-8326: MLCE: Update constraint message for AVERAGE_POOL_2D - Added missing constraint message for stride height by adding the constraint_stride_width_no_upper_limit to AVERAGE_POOL_2D Change-Id: Ib716fb19e44cb8735b52270b557998d4cbf5cb1c Signed-off-by: Johan Alfven --- ethosu/vela/tflite_supported_operators.py | 1 + 1 file changed, 1 insertion(+) (limited to 'ethosu') diff --git a/ethosu/vela/tflite_supported_operators.py b/ethosu/vela/tflite_supported_operators.py index 45003913..e5686f4c 100644 --- a/ethosu/vela/tflite_supported_operators.py +++ b/ethosu/vela/tflite_supported_operators.py @@ -254,6 +254,7 @@ class TFLiteSupportedOperators: self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_stride_range) # AVG pooling specific checks: for op_type in TFLiteSupportedOperators.avg_pooling_ops: + self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_stride_width_no_upper_limit) self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_stride_range_no_padding) self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_filter_range) self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_filter_height_range_valid_pad) -- cgit v1.2.1