From 69482271d3e02af950d2d0f1947ae6c3eeed537b Mon Sep 17 00:00:00 2001
From: James Conroy <james.conroy@arm.com>
Date: Fri, 19 Oct 2018 10:41:35 +0100
Subject: IVGCVSW-2024: Support NHWC for Pooling2D CpuRef

* Adds implementation to plumb DataLayout parameter
  for Pooling2D on CpuRef.
* Adds unit tests to execute Pooling2D on CpuRef
  using NHWC data layout.
* Refactors original tests to use DataLayoutIndexed
  and removes duplicate code.

Change-Id: Ife7e0861a886cf58a2042e5be20e5b27af4528c9
---
 include/armnn/Descriptors.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'include/armnn')

diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index a7eca43da3..648477e09b 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -194,7 +194,7 @@ struct Pooling2dDescriptor
     uint32_t            m_StrideY;
     OutputShapeRounding m_OutputShapeRounding;
     PaddingMethod       m_PaddingMethod;
-    DataLayout          m_DataLayout;
+    DataLayoutIndexed   m_DataLayout;
 };
 
 struct FullyConnectedDescriptor
-- 
cgit v1.2.1