From 061dd36716a17c1065de6143f61e41d03896d541 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Wed, 17 Oct 2018 17:10:27 +0100 Subject: COMPMID-1451: Fix allocation of weights in Deconvolution Change-Id: If3ca0b034a7448df1e5349b51a2b124f1b4e99c1 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/153956 Reviewed-by: Georgios Pinitas Tested-by: bsgcomp --- arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h | 2 +- arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) (limited to 'arm_compute/runtime') diff --git a/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h b/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h index 39cbe0cafa..7a58c5acef 100644 --- a/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h +++ b/arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h @@ -123,7 +123,7 @@ private: CLConvolutionLayer _conv_f; CPPFlipWeightsKernel _flip_weights; CLTensor _scaled_output; - ICLTensor *_weights; + ICLTensor *_original_weights; CLTensor _weights_flipped; bool _is_prepared; }; diff --git a/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h index 73870093b7..277945d617 100644 --- a/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h @@ -125,6 +125,7 @@ private: CPPFlipWeightsKernel _flip_weights; Tensor _scaled_output; Tensor _weights_flipped; + const ITensor *_original_weights; ITensor *_input; PadStrideInfo _info; std::pair _inner_border; -- cgit v1.2.1