diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2018-09-28 14:14:12 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:57 +0100 |
commit | bcd3c85b5a7657b38f503676b88a80ae74165acd (patch) | |
tree | a1f4f934aea17f0faa30f64b9eec0108bbe09ad3 /include/armnn/INetwork.hpp | |
parent | e1fa50c70c647e3ba6dec0eb49ce82a3820fb6bb (diff) | |
download | armnn-bcd3c85b5a7657b38f503676b88a80ae74165acd.tar.gz |
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
Diffstat (limited to 'include/armnn/INetwork.hpp')
-rw-r--r-- | include/armnn/INetwork.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp index 1d2679aabb..2c83909c83 100644 --- a/include/armnn/INetwork.hpp +++ b/include/armnn/INetwork.hpp @@ -228,9 +228,11 @@ public: /// Adds an L2 normalization layer to the network. /// Normalization is performed along dimension 1, but requires a 4d input. + /// @param desc - Parameters for the L2 normalization operation. /// @param name - Optional name for the layer. /// @return - Interface for configuring the layer. - virtual IConnectableLayer* AddL2NormalizationLayer(const char* name = nullptr) = 0; + virtual IConnectableLayer* AddL2NormalizationLayer(const L2NormalizationDescriptor& desc, + const char* name = nullptr) = 0; /// Adds a layer with no inputs and a single output, which always corresponds to /// the passed in constant tensor. |