diff options
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; } |