aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/SubGraph.hpp
diff options
context:
space:
mode:
authorMatteo Martincigh <matteo.martincigh@arm.com>2019-01-31 12:09:49 +0000
committerMatteo Martincigh <matteo.martincigh@arm.com>2019-01-31 17:25:45 +0000
commit0c051f9b6b8d7d1602e81d2977dda449b6392642 (patch)
tree0d9fbca37c82bd2bc8d5799c3708bc70392455d6 /src/armnn/SubGraph.hpp
parent4a55201f9d24d6bed46781ec56ba8862ddc7c26b (diff)
downloadarmnn-0c051f9b6b8d7d1602e81d2977dda449b6392642.tar.gz
IVGCVSW-2471 Code cleanup where necessary
* Added comments for the new API functions * Added comments to some critical parts of the code * Removed unnecessary code Change-Id: I04d4e7373b45b8a3806cacc93a8bd09351c0972a Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Diffstat (limited to 'src/armnn/SubGraph.hpp')
-rw-r--r--src/armnn/SubGraph.hpp7
1 files changed, 6 insertions, 1 deletions
diff --git a/src/armnn/SubGraph.hpp b/src/armnn/SubGraph.hpp
index 81166f1285..cb44f8ce3d 100644
--- a/src/armnn/SubGraph.hpp
+++ b/src/armnn/SubGraph.hpp
@@ -56,7 +56,7 @@ public:
/// 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 graph this sub-graph is a view of.
+ /// 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;
@@ -85,8 +85,13 @@ public:
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.