commit 1952622c4b9f2cecdc93dedf2d6a9a8a94eac312
author David Beck <david.beck@arm.com> Wed Sep 12 16:00:08 2018 +0100
committer Matthew Bentham <matthew.bentham@arm.com> Mon Oct 01 14:56:47 2018 +0100
tree aab1de02cc0adb6a4250fb94c869a0f8ded6fa40
parent c3b012e51f0351c4f74f33fbb68e0d4bd76a6ea9

IVGCVSW-1804 : add subtraction layer to the public interface

Change-Id: I2c1b622b5199000c01af387bb994267c2b91f89c

include/armnn/INetwork.hpp

diff --git a/include/armnn/INetwork.hpp b/include/armnn/INetwork.hpp
index be03f83..0405074 100644
--- a/include/armnn/INetwork.hpp
+++ b/include/armnn/INetwork.hpp
@@ -274,6 +274,11 @@
     /// @return - Interface for configuring the layer.
     virtual IConnectableLayer* AddDivisionLayer(const char* name = nullptr) = 0;
 
+    /// Adds a subtraction layer to the network.
+    /// @param name - Optional name for the layer.
+    /// @return - Interface for configuring the layer.
+    virtual IConnectableLayer* AddSubtractionLayer(const char* name = nullptr) = 0;
+
 protected:
     ~INetwork() {}
 };

src/armnn/Network.cpp

diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 507668a..dc531d1 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -589,6 +589,11 @@
     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))
 {

src/armnn/Network.hpp

diff --git a/src/armnn/Network.hpp b/src/armnn/Network.hpp
index 3582439..b6b8548 100644
--- a/src/armnn/Network.hpp
+++ b/src/armnn/Network.hpp
@@ -115,6 +115,8 @@
 
     IConnectableLayer* AddDivisionLayer(const char* name = nullptr) override;
 
+    IConnectableLayer* AddSubtractionLayer(const char* name = nullptr) override;
+
 private:
     IConnectableLayer* AddFullyConnectedLayerImpl(const FullyConnectedDescriptor& fullyConnectedDescriptor,
         const ConstTensor& weights,