aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2020-01-09 16:45:46 +0000
committerGeorgios Pinitas <georgios.pinitas@arm.com>2020-01-10 17:13:59 +0000
commitddc25f93cc0f4ae72cc00353ee494145194eed44 (patch)
tree9bca5c7edeb25012d4dcdb311a73db097bee8d61 /arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h
parent156208d1b5874f3fda428bf379d376ffb697c971 (diff)
downloadComputeLibrary-ddc25f93cc0f4ae72cc00353ee494145194eed44.tar.gz
COMPMID-2819: Retain layout during configuration for multiple functions.
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com> Change-Id: Ia528762dc5a93bebfd8fd037bf1f4e75d0b8a6de Reviewed-on: https://review.mlplatform.org/c/2566 Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h')
-rw-r--r--arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h3
1 files changed, 2 insertions, 1 deletions
diff --git a/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h b/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h
index 081b01aad3..faf97e45dc 100644
--- a/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h
+++ b/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2017-2019 ARM Limited.
+ * Copyright (c) 2017-2020 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -94,6 +94,7 @@ public:
const ICLTensor *_biases;
const ICLTensor *_weights;
ICLTensor *_output;
+ DataLayout _data_layout;
BorderSize _border_size;
int _conv_stride_x;
int _conv_stride_y;