aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2018-02-13 18:50:55 +0000
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:47:18 +0000
commit056b5d9dc6a87dc220792b7fe5ddea33692cc471 (patch)
tree91f46190f0b5df2ae8bee0e5a046340479c04f8e /src/runtime/NEON/functions/NEDeconvolutionLayer.cpp
parent69af6cf5ec6edce564dd5ef97baba5a1325e8763 (diff)
downloadComputeLibrary-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 'src/runtime/NEON/functions/NEDeconvolutionLayer.cpp')
-rw-r--r--src/runtime/NEON/functions/NEDeconvolutionLayer.cpp10
1 files changed, 3 insertions, 7 deletions
diff --git a/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp b/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp
index b293fa080a..193e8c3fca 100644
--- a/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp
+++ b/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp
@@ -33,13 +33,11 @@ using namespace arm_compute::misc::shape_calculator;
NEDeconvolutionLayer::NEDeconvolutionLayer(std::shared_ptr<IMemoryManager> memory_manager) // NOLINT
: _memory_group(std::move(memory_manager)),
- _direct_conv_f(),
_conv_f(),
_scaled_output(),
_input(nullptr),
_info(),
- _inner_border(),
- _run_direct_convolution(false)
+ _inner_border()
{
}
@@ -53,8 +51,6 @@ void NEDeconvolutionLayer::configure(ITensor *input, const ITensor *weights, con
_input = input;
_info = info;
_inner_border = std::make_pair(inner_border_right, inner_border_top);
- // FIXME: ConvolutionLayer Segfaults in GEMM assembly code for 1x1 convolutions
- _run_direct_convolution = (weights->info()->dimension(0) == weights->info()->dimension(1)) && (weights->info()->dimension(0) == 1);
const unsigned int stride_x = info.stride().first;
const unsigned int stride_y = info.stride().second;
@@ -78,7 +74,7 @@ void NEDeconvolutionLayer::configure(ITensor *input, const ITensor *weights, con
// setup the function to convolve the upscaled output
const PadStrideInfo conv_info(1, 1, 0, 0, 0, 0, DimensionRoundingType::CEIL);
- (_run_direct_convolution) ? _direct_conv_f.configure(&_scaled_output, weights, bias, output, conv_info) : _conv_f.configure(&_scaled_output, weights, bias, output, conv_info);
+ _conv_f.configure(&_scaled_output, weights, bias, output, conv_info);
// Allocate auxiliary tensors
_scaled_output.allocator()->allocate();
@@ -119,7 +115,7 @@ void NEDeconvolutionLayer::run()
}
// Run convolution layer
- (_run_direct_convolution) ? _direct_conv_f.run() : _conv_f.run();
+ _conv_f.run();
_memory_group.release();
}