diff options
author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-11-08 12:12:55 +0000 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-11-08 16:22:30 +0000 |
commit | 5e25b12bf292c1b9301a4a54cdca5d3c657a5d3a (patch) | |
tree | 499c49793bf393443629f321b4f8332749bd8198 /src/runtime/CL/functions | |
parent | 72f4ae5a53fe24226ff16ed9c339171887d74874 (diff) | |
download | ComputeLibrary-5e25b12bf292c1b9301a4a54cdca5d3c657a5d3a.tar.gz |
COMPMID-1451: Allow weights retention in CLDeconvolutionLayer
Change-Id: I953f3b63aa4910650a1a3f6faea31beb4f6f376a
Diffstat (limited to 'src/runtime/CL/functions')
-rw-r--r-- | src/runtime/CL/functions/CLDeconvolutionLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLDeconvolutionLayer.cpp b/src/runtime/CL/functions/CLDeconvolutionLayer.cpp index bbf4e66adb..e07feb226a 100644 --- a/src/runtime/CL/functions/CLDeconvolutionLayer.cpp +++ b/src/runtime/CL/functions/CLDeconvolutionLayer.cpp @@ -135,7 +135,7 @@ void CLDeconvolutionLayer::configure(ICLTensor *input, ICLTensor *weights, const // Perform validation step ARM_COMPUTE_ERROR_THROW_ON(CLDeconvolutionLayer::validate(input->info(), weights->info(), bias == nullptr ? nullptr : bias->info(), output->info(), info, inner_border_right, inner_border_top)); - _is_prepared = false; + _is_prepared = weights_info.retain_internal_weights(); _memory_group.manage(&_scaled_output); |