diff options
| author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-09 19:06:22 +0100 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-10 11:04:10 +0100 |
| commit | f3d102114a6f837f40400c4de50915abc488f3a5 (patch) | |
| tree | f524cdd3cd40f5d8df321a935988ed80c9f4a66b /src/armnn/Optimizer.cpp | |
| parent | 724e48013142562b7f09c9c819f57c314c4ee3d4 (diff) | |
| download | armnn-f3d102114a6f837f40400c4de50915abc488f3a5.tar.gz | |
IVGCVSW-3030 Added move operators to the Graph class
* Updated the LayerInGraph class to properly support
the new Reparent operation
* Improved the Graph class destruction process to take into
account eventual reparent layer operations
* Added new ForEachLayerInGraph utility function to safely
loop through all the layers in the graph
Change-Id: Ie67cbdee0c3c8625662ebfa00f860ae0d2fac59c
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Diffstat (limited to 'src/armnn/Optimizer.cpp')
| -rw-r--r-- | src/armnn/Optimizer.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/Optimizer.cpp b/src/armnn/Optimizer.cpp index 5e50c01c09..4d0aae8491 100644 --- a/src/armnn/Optimizer.cpp +++ b/src/armnn/Optimizer.cpp @@ -32,7 +32,9 @@ void Optimizer::Pass(Graph& graph, const Optimizations& optimizations) if ((*it)->IsOutputUnconnected()) { - it = graph.EraseLayer(it); + auto next = std::next(graph.GetPosInGraph(**it)); + graph.EraseLayer(it); + it = next; graphNeedsSorting = true; } |