From 1952622c4b9f2cecdc93dedf2d6a9a8a94eac312 Mon Sep 17 00:00:00 2001
From: David Beck <david.beck@arm.com>
Date: Wed, 12 Sep 2018 16:00:08 +0100
Subject: IVGCVSW-1804 : add subtraction layer to the public interface

Change-Id: I2c1b622b5199000c01af387bb994267c2b91f89c
---
 src/armnn/Network.cpp | 5 +++++
 src/armnn/Network.hpp | 2 ++
 2 files changed, 7 insertions(+)

(limited to 'src')

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 507668a7d9..dc531d10ac 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -589,6 +589,11 @@ IConnectableLayer* Network::AddDivisionLayer(const char* name)
     return m_Graph->AddLayer<DivisionLayer>(name);
 }
 
+IConnectableLayer* Network::AddSubtractionLayer(const char* name)
+{
+    return m_Graph->AddLayer<SubtractionLayer>(name);
+}
+
 OptimizedNetwork::OptimizedNetwork(std::unique_ptr<Graph> graph)
     : m_Graph(std::move(graph))
 {
diff --git a/src/armnn/Network.hpp b/src/armnn/Network.hpp
index 3582439426..b6b8548f08 100644
--- a/src/armnn/Network.hpp
+++ b/src/armnn/Network.hpp
@@ -115,6 +115,8 @@ public:
 
     IConnectableLayer* AddDivisionLayer(const char* name = nullptr) override;
 
+    IConnectableLayer* AddSubtractionLayer(const char* name = nullptr) override;
+
 private:
     IConnectableLayer* AddFullyConnectedLayerImpl(const FullyConnectedDescriptor& fullyConnectedDescriptor,
         const ConstTensor& weights,
-- 
cgit v1.2.1