diff options
author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-10-17 17:10:27 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:55:45 +0000 |
commit | 061dd36716a17c1065de6143f61e41d03896d541 (patch) | |
tree | 8b959d0da13357bd9d9f0f54c45ddf06cfa33e8f /arm_compute/runtime | |
parent | ed5a492ba791d8c8b3334749d4ae946b8f11d13d (diff) | |
download | ComputeLibrary-061dd36716a17c1065de6143f61e41d03896d541.tar.gz |
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 <georgios.pinitas@arm.com>
Tested-by: bsgcomp <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime')
-rw-r--r-- | arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h | 2 | ||||
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h | 1 |
2 files changed, 2 insertions, 1 deletions
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<unsigned int, unsigned int> _inner_border; |