aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/Descriptors.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r--include/armnn/Descriptors.hpp2
1 files changed, 2 insertions, 0 deletions
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