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/NEON/functions/NEDeconvolutionLayer.h | |
| 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/NEON/functions/NEDeconvolutionLayer.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h | 1 |
1 files changed, 1 insertions, 0 deletions
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; |