diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-03-01 16:03:50 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:50:48 +0000 |
| commit | 57dac8400d56a4b68975d5563a9540c96d49fe5f (patch) | |
| tree | cf474c6690c02115e623d5e7d867be665050f87e /arm_compute/runtime/NEON/functions/NEPoolingLayer.h | |
| parent | 0ef7e670a22248806458d7327db9e8b8c4db4ce6 (diff) | |
| download | ComputeLibrary-57dac8400d56a4b68975d5563a9540c96d49fe5f.tar.gz | |
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 <georgios.pinitas@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEPoolingLayer.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEPoolingLayer.h | 1 |
1 files changed, 1 insertions, 0 deletions
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__ */ |