From 885033b5bf2f6513b438f273b2bc71964f0c6c59 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Thu, 21 Jul 2022 11:46:03 +0100 Subject: MLBEDSW-4157: Add RESIZE_NEAREST_NEIGHBOR support - Changed ResizeBilinear to support ResizeNearestNeighbor as well for 1x1 IFM, IFM equal OFM, and non-align corners - Added support for ResizeNearestNeighbor with align corners by converting to a DepthwiseConv - Updated supported operator unit tests - Added is_resize() helper function and some associated refactoring Signed-off-by: Tim Hall Change-Id: Id5bdf2a25e8aa6a4f28b7236250abf768141ce37 --- ethosu/vela/pass_packing.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'ethosu/vela/pass_packing.py') diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 050b0965..988e52e6 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -61,10 +61,9 @@ mac_main_ops = set( Op.AvgPool, Op.MaxPool, Op.ReduceSum, - # deconvolution - Op.ResizeBilinear, ) -) + # resize ops use pooling operations unless explicitly converted to other operations prior to pass packing +) | Op.op_set(Op.is_resize_op) binary_elem_wise_main_ops = Op.op_set(Op.is_binary_elementwise_op) -- cgit v1.2.1