diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-01-31 12:09:49 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-01-31 17:25:45 +0000 |
commit | 0c051f9b6b8d7d1602e81d2977dda449b6392642 (patch) | |
tree | 0d9fbca37c82bd2bc8d5799c3708bc70392455d6 /src/armnn/SubGraph.hpp | |
parent | 4a55201f9d24d6bed46781ec56ba8862ddc7c26b (diff) | |
download | armnn-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.hpp | 7 |
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. |