From 33cea4db0b2729c5dbd50f9c0985578c60baffdd Mon Sep 17 00:00:00 2001 From: narpra01 Date: Thu, 27 Sep 2018 16:46:14 +0100 Subject: IVGCVSW-1919 - data layout parameter for Normalization Change-Id: I33dce72bb0f1e25425dc058d6213a7cdf56eecd2 --- include/armnn/Descriptors.hpp | 2 ++ 1 file changed, 2 insertions(+) (limited to 'include/armnn/Descriptors.hpp') diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 893db69700..bc1b59bdf5 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -262,6 +262,7 @@ struct NormalizationDescriptor , m_Alpha(0.f) , m_Beta(0.f) , m_K(0.f) + , m_DataLayout(DataLayout::NCHW) {} NormalizationAlgorithmChannel m_NormChannelType; @@ -270,6 +271,7 @@ struct NormalizationDescriptor float m_Alpha; float m_Beta; float m_K; + DataLayout m_DataLayout; }; struct BatchNormalizationDescriptor -- cgit v1.2.1