diff options
Diffstat (limited to 'src/armnn/Network.cpp')
| -rw-r--r-- | src/armnn/Network.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 4f5e2974ed..49e60e1ad4 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -468,9 +468,10 @@ resizeDescriptor, const char* name) return m_Graph->AddLayer<ResizeBilinearLayer>(resizeDescriptor,name); } -IConnectableLayer* Network::AddL2NormalizationLayer(const char* name) +IConnectableLayer* Network::AddL2NormalizationLayer(const L2NormalizationDescriptor& desc, + const char* name) { - return m_Graph->AddLayer<L2NormalizationLayer>(name); + return m_Graph->AddLayer<L2NormalizationLayer>(desc, name); } IConnectableLayer* Network::AddConstantLayer(const ConstTensor& input, const char* name) |