diff options
author | David Monahan <david.monahan@arm.com> | 2019-05-10 11:52:14 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-10 13:09:20 +0100 |
commit | 5200afa2742ad9cd1cda7fbce8604794c0616818 (patch) | |
tree | c12a889ebceff8618b672876a353ec13e5c4291d /src/armnn/Graph.hpp | |
parent | afe78896447a90e065df6d9b1e8753bd85525750 (diff) | |
download | armnn-5200afa2742ad9cd1cda7fbce8604794c0616818.tar.gz |
IVGCVSW-3034 Updates to SubstituteSubGraph and
ReplaceSubgraphConnections to support Graphs instead of
SubGraphViews
* Added layer iteration function to SubgraphView similar to
the Graph's one
* Updated SubstituteSubgraph to reparent the layers to the
calling graph
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: Ib2f8e70decca4a59c53ceb127e07ef5a430d1005
Diffstat (limited to 'src/armnn/Graph.hpp')
-rw-r--r-- | src/armnn/Graph.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/Graph.hpp b/src/armnn/Graph.hpp index c5b1b045a7..47e0e3b317 100644 --- a/src/armnn/Graph.hpp +++ b/src/armnn/Graph.hpp @@ -36,7 +36,7 @@ public: } template <typename Func> - void ForEachLayerInGraph(Func func) + void ForEachLayer(Func func) const { for (auto it = m_Layers.begin(); it != m_Layers.end(); ) { @@ -110,7 +110,7 @@ public: m_LayersInOrder = std::move(other.m_LayersInOrder); m_Views = std::move(other.m_Views); - other.ForEachLayerInGraph([this](Layer* otherLayer) + other.ForEachLayer([this](Layer* otherLayer) { otherLayer->Reparent(*this, m_Layers.end()); }); @@ -123,7 +123,7 @@ public: ~Graph() { - ForEachLayerInGraph([](Layer* layer) + ForEachLayer([](Layer* layer) { delete layer; }); |