aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/runtime')
-rw-r--r--arm_compute/runtime/CL/functions/CLDeconvolutionLayer.h2
-rw-r--r--arm_compute/runtime/NEON/functions/NEDeconvolutionLayer.h1
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;