From 33a6990ee6ba7bf85b88822d9723060262d00785 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Fri, 23 Feb 2018 15:01:52 +0000 Subject: COMPMID-540 Replace NEDeconvolutionLayerUpsampleKernel with NEScaleKernel Change-Id: Ic29557cca24447ef40fa2cfca84f208b4d43f8de Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/122180 Tested-by: Jenkins Reviewed-by: Michele DiGiorgio Reviewed-by: Pablo Tello --- arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h | 2 ++ 1 file changed, 2 insertions(+) (limited to 'arm_compute/runtime/NEON/functions') diff --git a/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h index 7566131488..c148c3ff69 100644 --- a/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h @@ -24,6 +24,7 @@ #ifndef __ARM_COMPUTE_NEDECONVOLUTIONLAYER_H__ #define __ARM_COMPUTE_NEDECONVOLUTIONLAYER_H__ +#include "arm_compute/runtime/CPP/functions/CPPUpsample.h" #include "arm_compute/runtime/NEON/functions/NEConvolutionLayer.h" #include "arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h" @@ -97,6 +98,7 @@ public: private: MemoryGroup _memory_group; NEConvolutionLayer _conv_f; + CPPUpsample _upsample_f; Tensor _scaled_output; ITensor *_input; PadStrideInfo _info; -- cgit v1.2.1