From 57dac8400d56a4b68975d5563a9540c96d49fe5f Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Thu, 1 Mar 2018 16:03:50 +0000 Subject: COMPMID-806 Add NHWC data format support format for NEON pooling Change-Id: I7ab174c72f3d56134fcec259a137739061fd12e9 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/123065 Reviewed-by: Georgios Pinitas Tested-by: Jenkins --- arm_compute/runtime/NEON/functions/NEPoolingLayer.h | 1 + 1 file changed, 1 insertion(+) (limited to 'arm_compute/runtime/NEON/functions/NEPoolingLayer.h') diff --git a/arm_compute/runtime/NEON/functions/NEPoolingLayer.h b/arm_compute/runtime/NEON/functions/NEPoolingLayer.h index 3ac0844798..4224f75c77 100644 --- a/arm_compute/runtime/NEON/functions/NEPoolingLayer.h +++ b/arm_compute/runtime/NEON/functions/NEPoolingLayer.h @@ -72,6 +72,7 @@ private: NEPoolingLayerKernel _pooling_layer_kernel; NEFillBorderKernel _border_handler; bool _is_global_pooling_layer; + DataLayout _data_layout; }; } #endif /* __ARM_COMPUTE_NEPOOLINGLAYER_H__ */ -- cgit v1.2.1