IVGCVSW-1896 : SubGraph selector

Change-Id: Iae9a76b10d84d7ba80136b54355f8d37d2df475d

1 files changed, 72 insertions, 0 deletions

diff --git a/src/armnn/SubGraph.cpp b/src/armnn/SubGraph.cpp
new file mode 100644
index 0000000000..5d41f32932
--- /dev/null
+++ b/src/armnn/SubGraph.cpp
@@ -0,0 +1,72 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "Layer.hpp"
+#include "SubGraph.hpp"
+
+#include <boost/numeric/conversion/cast.hpp>
+
+namespace armnn
+{
+
+SubGraph::SubGraph()
+{
+}
+
+SubGraph::SubGraph(InputSlots && inputs,
+                   OutputSlots && outputs,
+                   Layers && layers)
+: m_InputSlots{inputs}
+, m_OutputSlots{outputs}
+, m_Layers{layers}
+{
+}
+
+const SubGraph::InputSlots & SubGraph::GetInputSlots() const
+{
+    return m_InputSlots;
+}
+
+const SubGraph::OutputSlots & SubGraph::GetOutputSlots() const
+{
+    return m_OutputSlots;
+}
+
+const InputSlot* SubGraph::GetInputSlot(unsigned int index) const
+{
+    return m_InputSlots.at(index);
+}
+
+InputSlot* SubGraph::GetInputSlot(unsigned int index)
+{
+    return  m_InputSlots.at(index);
+}
+
+const OutputSlot* SubGraph::GetOutputSlot(unsigned int index) const
+{
+    return m_OutputSlots.at(index);
+}
+
+OutputSlot* SubGraph::GetOutputSlot(unsigned int index)
+{
+    return m_OutputSlots.at(index);
+}
+
+unsigned int SubGraph::GetNumInputSlots() const
+{
+    return boost::numeric_cast<unsigned int>(m_InputSlots.size());
+}
+
+unsigned int SubGraph::GetNumOutputSlots() const
+{
+    return boost::numeric_cast<unsigned int>(m_OutputSlots.size());
+}
+
+const SubGraph::Layers & SubGraph::GetLayers() const
+{
+    return m_Layers;
+}
+
+} // namespace armnn