From 9a03fdff316662be69a1adc4e391e43bc6519b08 Mon Sep 17 00:00:00 2001 From: Charles Xu Date: Thu, 2 Jul 2020 15:12:40 +0200 Subject: MLBEDSW-2569:Support 1x1 IFM ResizeBilinear Signed-off-by: Charles Xu Change-Id: I44428d77b2e8e44a477e5c4dfe28ab8dd1792838 --- ethosu/vela/supported_operators.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'ethosu/vela/supported_operators.py') diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py index e8e8d852..3ec3429a 100644 --- a/ethosu/vela/supported_operators.py +++ b/ethosu/vela/supported_operators.py @@ -236,6 +236,8 @@ class SupportedOperators: def check_resize_restrictions(self, op): # check unsupported upscaling factor if op.type == "ResizeBilinear": + if op.inputs[0].shape[1] == 1 and op.inputs[0].shape[2] == 1: + return True upscaled_shape = [op.inputs[0].shape[1] * 2, op.inputs[0].shape[2] * 2] out_shape = op.outputs[0].shape[1:3] if not op.attrs["align_corners"] and out_shape != upscaled_shape: -- cgit v1.2.1