From f5a80e18badddf2fc61a89a71f822dfd4169cda4 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 15 Jan 2020 17:48:20 +0000 Subject: COMPMID-2819: Retain layout in NEIm2ColKernel Signed-off-by: Georgios Pinitas Change-Id: I5af7f9f87cedcfc8a4d59e0376113d2f285754fd Reviewed-on: https://review.mlplatform.org/c/2592 Reviewed-by: Pablo Marquez Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- arm_compute/core/NEON/kernels/NEIm2ColKernel.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'arm_compute/core') diff --git a/arm_compute/core/NEON/kernels/NEIm2ColKernel.h b/arm_compute/core/NEON/kernels/NEIm2ColKernel.h index f76521f770..689da857a7 100644 --- a/arm_compute/core/NEON/kernels/NEIm2ColKernel.h +++ b/arm_compute/core/NEON/kernels/NEIm2ColKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -131,6 +131,7 @@ private: unsigned int _kernel_height; bool _has_bias; Size2D _dilation; + DataLayout _data_layout; }; } // namespace arm_compute #endif /*__ARM_COMPUTE_NEIM2COLKERNEL_H__ */ -- cgit v1.2.1