diff options
Diffstat (limited to 'ethosu/vela/tflite_supported_operators.py')
| -rw-r--r-- | ethosu/vela/tflite_supported_operators.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_supported_operators.py b/ethosu/vela/tflite_supported_operators.py index d590054f..a3c0dd8d 100644 --- a/ethosu/vela/tflite_supported_operators.py +++ b/ethosu/vela/tflite_supported_operators.py @@ -162,6 +162,7 @@ class TFLiteSupportedOperators: # Resizing specific checks: for op_type in TFLiteSupportedOperators.resizing_ops: self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_resize) + self.specific_constraints[op_type].append(TFLiteSupportedOperators.constraint_bilinear_resize_attrs) # Vector Product specific checks: for op_type in TFLiteSupportedOperators.fc_vector_products: @@ -530,6 +531,14 @@ class TFLiteSupportedOperators: return valid, f"Op has ifm_shape={ifm_shape}, ofm_shape={ofm_shape} and align_corners={align_corners}" @staticmethod + def constraint_bilinear_resize_attrs(op): + "half_pixel_centers are not supported" + valid = True + if op.attrs.get("half_pixel_centers"): + valid = False + return valid, f"Op has half_pixel_centers set to {not valid}." + + @staticmethod def constraint_pad_shape(op): "The padding tensor must have the shape [3,2] or [4,2]" valid = op.inputs[1].shape in ([3, 2], [4, 2]) |