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 /src/armnn/layers/L2NormalizationLayer.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 'src/armnn/layers/L2NormalizationLayer.hpp')
| -rw-r--r-- | src/armnn/layers/L2NormalizationLayer.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/layers/L2NormalizationLayer.hpp b/src/armnn/layers/L2NormalizationLayer.hpp index fdf46eb921..bf4d49e7c8 100644 --- a/src/armnn/layers/L2NormalizationLayer.hpp +++ b/src/armnn/layers/L2NormalizationLayer.hpp @@ -4,12 +4,12 @@ // #pragma once -#include <Layer.hpp> +#include "LayerWithParameters.hpp" namespace armnn { -class L2NormalizationLayer : public Layer +class L2NormalizationLayer : public LayerWithParameters<L2NormalizationDescriptor> { public: virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph& graph, @@ -20,7 +20,7 @@ public: void ValidateTensorShapesFromInputs() override; protected: - L2NormalizationLayer(const char* name); + L2NormalizationLayer(const L2NormalizationDescriptor& param, const char* name); ~L2NormalizationLayer() = default; }; |