aboutsummaryrefslogtreecommitdiff
path: root/include
diff options
context:
space:
mode:
authornarpra01 <narumol.prangnawarat@arm.com>2018-09-27 16:46:14 +0100
committerMatthew Bentham <matthew.bentham@arm.com>2018-10-10 16:16:57 +0100
commit33cea4db0b2729c5dbd50f9c0985578c60baffdd (patch)
treeecea0ce825b52a2225788362c76f40f6ec548cda /include
parent0dbe0ee25312b728d77383d11c465156e64ae757 (diff)
downloadarmnn-33cea4db0b2729c5dbd50f9c0985578c60baffdd.tar.gz
IVGCVSW-1919 - data layout parameter for Normalization
Change-Id: I33dce72bb0f1e25425dc058d6213a7cdf56eecd2
Diffstat (limited to 'include')
-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