From 616acc602eff2a4a56a8f53b160eed51eaaf0afd Mon Sep 17 00:00:00 2001 From: Rob Hughes Date: Wed, 24 Jul 2019 08:21:03 +0000 Subject: NNXSW-1697 Remove use of SubgraphView that may become invalid. The mainSubgraph variable held references to layers in the optimised graph that become invalid when a graph substitution is made. The fix is to always use the optimised graph itself rather than a view of it. The view served no purpose anyway. !android-nn-driver:1221 Change-Id: I579f503827db8f7fcb01443237c9d888d496425f Signed-off-by: Rob Hughes --- src/armnn/Network.cpp | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index b80e0e7eec..0d06042abe 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -334,9 +334,6 @@ OptimizationResult ApplyBackendOptimizations(OptimizedNetwork* optNetObjPtr, // Get the optimized graph Graph& optGraph = optNetObjPtr->GetGraph(); - // Get the entire graph as a sub-graph - SubgraphView mainSubgraph(optGraph); - // Run backend specific optimizations auto const& backendRegistry = BackendRegistryInstance(); for (auto&& selectedBackend : backendSettings.m_SelectedBackends) @@ -347,7 +344,7 @@ OptimizationResult ApplyBackendOptimizations(OptimizedNetwork* optNetObjPtr, // Select sub-graphs based on backend SubgraphViewSelector::Subgraphs subgraphs = - SubgraphViewSelector::SelectSubgraphs(mainSubgraph, + SubgraphViewSelector::SelectSubgraphs(optGraph, // Select layers assigned to the requested backend [&backendObjPtr](const Layer& layer) { -- cgit v1.2.1