From 8eb675eb77865b5d2491f5b2d650ce993cab738c Mon Sep 17 00:00:00 2001
From: Matteo Martincigh <matteo.martincigh@arm.com>
Date: Wed, 17 Oct 2018 14:43:29 +0100
Subject: IVGCVSW-2038 + IVGCVSW-2039 + IVGCVSW-2040 Add NHWC support to the
 Float32 and UInt8 BatchNormalization workloads

 * Enabled NHWC support in RefBatchNormalizationFloat32Workload
 * Added NHWC unit tests for both FP32 and U8
 * Refactored the existing unit tests

Change-Id: I6aa18f1dcc0666b80a17a7ed229cf53607bae147
---
 include/armnn/Descriptors.hpp | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'include')

diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index c0510055f2..a5b1d64732 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -289,9 +289,11 @@ struct BatchNormalizationDescriptor
 {
     BatchNormalizationDescriptor()
     : m_Eps(0.0001f)
+    , m_DataLayout(DataLayout::NCHW)
     {}
 
     float m_Eps;
+    DataLayout m_DataLayout;
 };
 
 struct FakeQuantizationDescriptor
-- 
cgit v1.2.1