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
committerMichele Di Giorgio <michele.digiorgio@arm.com>2020-01-10 16:58:20 +0000
commit7fdcfb113cc5651eeb5f35333a41434bc4c2223f (patch)
tree3a5cfaed83f8bb9ca88b0fdf44c263088338dc4f /arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h
parent07958785a2542bf461be48241127d4ad5044d08a (diff)
downloadComputeLibrary-7fdcfb113cc5651eeb5f35333a41434bc4c2223f.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 fef0707190..b37d3a4047 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;