From b609c93e81b1ff60764c5b724f4f5511215c84d6 Mon Sep 17 00:00:00 2001 From: Annop Wongwathanarat Date: Mon, 16 Jan 2023 14:36:45 +0000 Subject: Skip upsampling for deconvolution when not needed If the input tensor's stride is 1 and the kernel size is 1x1, skip upsampling step and pass the input tensor pointer for convolution directly. Partially resolve: [ONCPUML-1137] Change-Id: I9de9444ff99cf35d44a51ccbe0fa6facc1035d27 Signed-off-by: Annop Wongwathanarat Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8994 Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Reviewed-by: Gunes Bayir Benchmark: Arm Jenkins --- arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h | 1 + 1 file changed, 1 insertion(+) (limited to 'arm_compute') diff --git a/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h index 15124d6041..869df69f11 100644 --- a/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h @@ -148,6 +148,7 @@ private: ITensor *_input; PadStrideInfo _info; bool _is_prepared; + bool _do_upsampling; }; } // arm_compute #endif /* ARM_COMPUTE_NEDECONVOLUTIONLAYER_H */ -- cgit v1.2.1