1 files changed, 109 insertions, 0 deletions
diff --git a/src/armnn/SubgraphView.hpp b/src/armnn/SubgraphView.hpp
new file mode 100644
index 0000000000..b211ad4046
--- /dev/null
+++ b/src/armnn/SubgraphView.hpp
@@ -0,0 +1,109 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include "Layer.hpp"
+#include "Graph.hpp"
+
+#include <vector>
+#include <list>
+
+namespace armnn
+{
+
+///
+/// The SubgraphView class represents a subgraph of a Graph.
+/// The data it holds, points to data held by layers of the Graph, so the
+/// the contents of the SubgraphView becomes invalid when the Layers are destroyed
+/// or changed.
+///
+class SubgraphView final
+{
+public:
+    using SubgraphViewPtr = std::unique_ptr<SubgraphView>;
+    using InputSlots = std::vector<InputSlot*>;
+    using OutputSlots = std::vector<OutputSlot*>;
+    using Layers = std::list<Layer*>;
+    using Iterator = Layers::iterator;
+    using ConstIterator = Layers::const_iterator;
+
+    /// Empty subgraphs are not allowed, they must at least have a parent graph.
+    SubgraphView() = delete;
+
+    /// Constructs a sub-graph from the entire given graph.
+    SubgraphView(Graph& graph);
+
+    /// Constructs a sub-graph with the given arguments and binds it to the specified parent graph.
+    SubgraphView(Graph* parentGraph, InputSlots&& inputs, OutputSlots&& outputs, Layers&& layers);
+
+    /// Constructs a sub-graph with the given arguments and uses the specified sub-graph to get a reference
+    /// to the parent graph.
+    SubgraphView(const SubgraphView& referenceSubgraph, InputSlots&& inputs, OutputSlots&& outputs, Layers&& layers);
+
+    /// Copy-constructor.
+    SubgraphView(const SubgraphView& subgraph);
+
+    /// Move-constructor.
+    SubgraphView(SubgraphView&& subgraph);
+
+    /// Constructs a sub-graph with only the given layer and uses the specified sub-graph to get a reference
+    /// to the parent graph.
+    SubgraphView(const SubgraphView& referenceSubgraph, IConnectableLayer* layer);
+
+    /// Updates this sub-graph with the contents of the whole given graph.
+    void Update(Graph& graph);
+
+    /// Adds a new layer, of type LayerType, to the parent graph of this sub-graph.
+    template <typename LayerT, typename... Args>
+    LayerT* AddLayer(Args&&... args) const;
+
+    const InputSlots& GetInputSlots() const;
+    const OutputSlots& GetOutputSlots() const;
+    const Layers& GetLayers() const;
+
+    const InputSlot* GetInputSlot(unsigned int index) const;
+    InputSlot* GetInputSlot(unsigned int index);
+
+    const OutputSlot* GetOutputSlot(unsigned int index) const;
+    OutputSlot* GetOutputSlot(unsigned int index);
+
+    unsigned int GetNumInputSlots() const;
+    unsigned int GetNumOutputSlots() const;
+
+    Iterator begin();
+    Iterator end();
+
+    ConstIterator begin() const;
+    ConstIterator end() const;
+
+    ConstIterator cbegin() const;
+    ConstIterator cend() const;
+
+private:
+    void CheckSubgraph();
+
+    /// The list of pointers to the input slots of the parent graph.
+    InputSlots m_InputSlots;
+
+    /// The list of pointers to the output slots of the parent graph.
+    OutputSlots m_OutputSlots;
+
+    /// The list of pointers to the layers of the parent graph.
+    Layers m_Layers;
+
+    /// Pointer to the graph this sub-graph is a view of.
+    Graph* m_ParentGraph;
+};
+
+template <typename LayerT, typename... Args>
+LayerT* SubgraphView::AddLayer(Args&&... args) const
+{
+    BOOST_ASSERT(m_ParentGraph);
+
+    return m_ParentGraph->AddLayer<LayerT>(args...);
+}
+
+} // namespace armnn