From 5e25b12bf292c1b9301a4a54cdca5d3c657a5d3a Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Thu, 8 Nov 2018 12:12:55 +0000 Subject: COMPMID-1451: Allow weights retention in CLDeconvolutionLayer Change-Id: I953f3b63aa4910650a1a3f6faea31beb4f6f376a --- src/runtime/CL/functions/CLDeconvolutionLayer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/runtime/CL') 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); -- cgit v1.2.1