diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-01 10:31:27 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-03 14:35:34 +0000 |
commit | 602af090a6b7285dc1041e3d8424836e5fa12cef (patch) | |
tree | 8ffccd841a911164a022e98e57630ba348ceb796 /src/armnn/Network.cpp | |
parent | 59e15b00ea51fee4baeea750dc11ab1952dfab1d (diff) | |
download | armnn-602af090a6b7285dc1041e3d8424836e5fa12cef.tar.gz |
IVGCVSW-3029 Remove any AddLayer capabilities from SubgraphView
* Removed the reference to the parent graph in SubgraphView
* Removed the AddLayer method in SubgraphView
* Updated the code where necessary to adapt to the new changes in
SubgraphView
* Fixed a check in the CreatePreCompiledWorkloadTest test function
Change-Id: I4d3af87f11ec3cd8f18a21b250a2d295da56e1a0
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
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 { |