diff options
Diffstat (limited to 'src/armnn/Network.cpp')
| -rw-r--r-- | src/armnn/Network.cpp | 17 |
1 files changed, 7 insertions, 10 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 9ef0c568a3..1047567cc4 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -348,13 +348,13 @@ OptimizationResult ApplyBackendOptimizations(OptimizedNetwork* optNetObjPtr, // Select sub-graphs based on backend SubgraphViewSelector::Subgraphs subgraphs = SubgraphViewSelector::SelectSubgraphs(mainSubgraph, - // Select layers assigned to the requested backend - [&backendObjPtr](const Layer& layer) - { - return layer.GetType() != LayerType::Input && - layer.GetType() != LayerType::Output && - layer.GetBackendId() == backendObjPtr->GetId(); - }); + // Select layers assigned to the requested backend + [&backendObjPtr](const Layer& layer) + { + return layer.GetType() != LayerType::Input && + layer.GetType() != LayerType::Output && + layer.GetBackendId() == backendObjPtr->GetId(); + }); if (subgraphs.empty()) { // No sub-graphs found, try with next selected backend @@ -388,9 +388,6 @@ OptimizationResult ApplyBackendOptimizations(OptimizedNetwork* optNetObjPtr, BOOST_ASSERT(l); l->SetBackendId(selectedBackend); }); - - // Recreate the sub-graph representing the entire graph - mainSubgraph.Update(optGraph); } else { |