diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-02-23 15:01:52 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:48:33 +0000 |
commit | 33a6990ee6ba7bf85b88822d9723060262d00785 (patch) | |
tree | 104e1b614b79a891e78fb7796a91c1981b4d6dd1 /arm_compute/runtime/NEON | |
parent | 6a8eb0c9c8d3f71ed4e718464c6d22ca0ffbbdd8 (diff) | |
download | ComputeLibrary-33a6990ee6ba7bf85b88822d9723060262d00785.tar.gz |
COMPMID-540 Replace NEDeconvolutionLayerUpsampleKernel with NEScaleKernel
Change-Id: Ic29557cca24447ef40fa2cfca84f208b4d43f8de
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/122180
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele DiGiorgio <michele.digiorgio@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h | 2 |
1 files changed, 2 insertions, 0 deletions
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; |