MLBEDSW-6928: Add int16 support for Resize Bilinear HPC
Setting bias tensor dtype to DataType.int32 solves rounding issues for
RB HPC int16.
Removing the input data type check also solves the issue of resize
nearest neighbor int16 ops incorrectly getting placed on the CPU.
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: Iee352bcb78e581c0cde3c203dfbe866f1f6fae18
diff --git a/ethosu/vela/tflite_supported_operators.py b/ethosu/vela/tflite_supported_operators.py
index 9aa174d..8b448df 100644
--- a/ethosu/vela/tflite_supported_operators.py
+++ b/ethosu/vela/tflite_supported_operators.py
@@ -666,12 +666,12 @@
@staticmethod
def constraint_resize_half_pixel_centers(op):
- """Half_pixel_centers are only supported for resize bilinear with IFM dtype int8 or uint8"""
- valid = op.ifm.dtype in (DataType.int8, DataType.uint8)
+ """Half_pixel_centers are only supported for resize bilinear"""
+ valid = True
half_pixel_centers = op.attrs.get("half_pixel_centers", False)
if half_pixel_centers and op.type != Op.ResizeBilinear:
valid = False
- return valid, f"Op type={op.type}, ifm dtype={op.ifm.dtype} and half_pixel_centers={half_pixel_centers}"
+ return valid, f"Op type={op.type} and half_pixel_centers={half_pixel_centers}"
@staticmethod
def constraint_resizebi_half_pixel_centers_dims(op):