From 14d9d986c63eb8f07acec2f075383f9f4843ae1c Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 13 Dec 2019 12:33:09 +0000 Subject: COMPMID-2819: Retain configuration step data layout to avoid side-effects. Configuring functions serially can lead to side-effects in tensor attributes. One of them is the data layout changing in case functions share same IO tensors. Retain DataLayout used during configuration. Change-Id: I17538ce08b86df6986f0fcf21fa6544fbd5bd74b Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/c/2470 Tested-by: Arm Jenkins Reviewed-by: Giorgio Arena --- arm_compute/core/NEON/kernels/NEPoolingLayerKernel.h | 1 + 1 file changed, 1 insertion(+) (limited to 'arm_compute/core/NEON/kernels/NEPoolingLayerKernel.h') diff --git a/arm_compute/core/NEON/kernels/NEPoolingLayerKernel.h b/arm_compute/core/NEON/kernels/NEPoolingLayerKernel.h index 47421eae9a..b36e10cd40 100644 --- a/arm_compute/core/NEON/kernels/NEPoolingLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NEPoolingLayerKernel.h @@ -194,6 +194,7 @@ private: const ITensor *_input; ITensor *_output; PoolingLayerInfo _pool_info; + DataLayout _data_layout; unsigned int _num_elems_processed_per_iteration; BorderSize _border_size; bool _is_square; -- cgit v1.2.1