diff options
| author | Diego Lopez Recas <Diego.LopezRecas@arm.com> | 2017-11-28 16:44:52 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:41:58 +0000 |
| commit | fa0add17fd08f623359fa19abfbcc681bbd8f220 (patch) | |
| tree | 048ba57e193ea9394175fa06c257f5f5db9598db /src | |
| parent | 4db041ec8adc565452b528fd36f5534999ccdfd0 (diff) | |
| download | ComputeLibrary-fa0add17fd08f623359fa19abfbcc681bbd8f220.tar.gz | |
Fix border handler for CLDepthwiseConvolution3x3
Change-Id: Ibf855a8ff86750978ecb81441369c83bc766f143
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/110987
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: BSG Visual Compute Jenkins server to access repositories on http://mpd-gerrit.cambridge.arm.com <bsgcomp@arm.com>
Diffstat (limited to 'src')
| -rw-r--r-- | src/runtime/CL/functions/CLDepthwiseConvolution.cpp | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLDepthwiseConvolution.cpp b/src/runtime/CL/functions/CLDepthwiseConvolution.cpp index baa05b921a..a701391c44 100644 --- a/src/runtime/CL/functions/CLDepthwiseConvolution.cpp +++ b/src/runtime/CL/functions/CLDepthwiseConvolution.cpp @@ -43,7 +43,14 @@ void CLDepthwiseConvolution3x3::configure(ICLTensor *input, const ICLTensor *wei _kernel.set_target(CLScheduler::get().target()); _kernel.configure(input, weights, biases, output, conv_info); - _border_handler.configure(input, _kernel.border_size(), BorderMode::CONSTANT, PixelValue(0)); + + // Configure border handler + PixelValue &&zero_value(0.f); + if(is_data_type_quantized_asymmetric(input->info()->data_type())) + { + zero_value = PixelValue(static_cast<uint8_t>(input->info()->quantization_info().offset)); + } + _border_handler.configure(input, _kernel.border_size(), BorderMode::CONSTANT, zero_value); } void CLDepthwiseConvolution3x3::run() |