From bcd3c85b5a7657b38f503676b88a80ae74165acd Mon Sep 17 00:00:00 2001 From: Matteo Martincigh Date: Fri, 28 Sep 2018 14:14:12 +0100 Subject: IVGCVSW-1863 Support NHWC for L2Normalization * Added L2NormalizationDescriptor struct with m_DataLyaout member * Updated all IsL2NormalizationSupported calls to take a descriptor as an argument * Updated L2NormalizationLayer to take a descriptor as an argument !android-nn-driver:150116 Change-Id: I0459352d19cfd269bc864a70cf73910bf44fdc01 --- src/armnn/Network.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/armnn/Network.hpp') diff --git a/src/armnn/Network.hpp b/src/armnn/Network.hpp index ea4284e38f..129513d657 100644 --- a/src/armnn/Network.hpp +++ b/src/armnn/Network.hpp @@ -98,7 +98,8 @@ public: IConnectableLayer* AddResizeBilinearLayer(const ResizeBilinearDescriptor& resizeDesc, const char* name = nullptr) override; - IConnectableLayer* AddL2NormalizationLayer(const char* name = nullptr) override; + IConnectableLayer* AddL2NormalizationLayer(const L2NormalizationDescriptor& desc, + const char* name = nullptr) override; IConnectableLayer* AddConstantLayer(const ConstTensor& input, const char* name = nullptr) override; -- cgit v1.2.1