commit 9a03fdff316662be69a1adc4e391e43bc6519b08
author Charles Xu <charles.xu@arm.com> Thu Jul 02 15:12:40 2020 +0200
committer Fredrik Knutsson <fredrik.knutsson.hunnebo@gmail.com> Wed Aug 05 10:45:33 2020 +0000
tree 6608df66254a762da1660693ee36481e76e97b8c
parent 0430882a652368d1da1638114d4fbf8c8f7c8d7b

MLBEDSW-2569:Support 1x1 IFM ResizeBilinear

Signed-off-by: Charles Xu <charles.xu@arm.com>
Change-Id: I44428d77b2e8e44a477e5c4dfe28ab8dd1792838

ethosu/vela/supported_operators.py

diff --git a/ethosu/vela/supported_operators.py b/ethosu/vela/supported_operators.py
index e8e8d85..3ec3429 100644
--- a/ethosu/vela/supported_operators.py
+++ b/ethosu/vela/supported_operators.py
@@ -236,6 +236,8 @@
     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: