diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-02-13 18:50:55 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:47:18 +0000 |
| commit | 056b5d9dc6a87dc220792b7fe5ddea33692cc471 (patch) | |
| tree | 91f46190f0b5df2ae8bee0e5a046340479c04f8e /arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h | |
| parent | 69af6cf5ec6edce564dd5ef97baba5a1325e8763 (diff) | |
| download | ComputeLibrary-056b5d9dc6a87dc220792b7fe5ddea33692cc471.tar.gz | |
COMPMID-765: Switch 1x1 DeconvolutionLayer to use the ConvolutionLayer
-Swithes the 1x1 DeconvolutionLayer to use the ConvolutionLayer instead
of the DirectConvolutionLayer.
Change-Id: I3ffe152c42c3b1c7ea572f264cd3215df01aedc2
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/120292
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h index 1b3297e8d0..7566131488 100644 --- a/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h @@ -95,14 +95,12 @@ public: void run() override; private: - MemoryGroup _memory_group; - NEDirectConvolutionLayer _direct_conv_f; - NEConvolutionLayer _conv_f; - Tensor _scaled_output; - ITensor *_input; - PadStrideInfo _info; + MemoryGroup _memory_group; + NEConvolutionLayer _conv_f; + Tensor _scaled_output; + ITensor *_input; + PadStrideInfo _info; std::pair<unsigned int, unsigned int> _inner_border; - bool _run_direct_convolution; }; } // arm_compute #endif /* __ARM_COMPUTE_NEDECONVOLUTIONLAYER_H__ */ |