diff options
| author | Cathal Corbett <cathal.corbett@arm.com> | 2021-12-15 17:12:59 +0000 |
|---|---|---|
| committer | Cathal Corbett <cathal.corbett@arm.com> | 2021-12-23 13:21:22 +0000 |
| commit | cbfd718464b8ac41f0338ae6565d8213d24c0a2a (patch) | |
| tree | f26da835108a0ed52ac0ffc8f7ebec64827b5033 /src/armnn/test/SubgraphViewTests.cpp | |
| parent | 81edc6217f76953c0be4c47f3d005cf48772ccb7 (diff) | |
| download | armnn-cbfd718464b8ac41f0338ae6565d8213d24c0a2a.tar.gz | |
IVGCVSW-6632 OptimizationViews: has INetwork rather than Graph for holding layers
* Deprecate the GetGraph() function in OptimizationViews & remove/fix
occurances where OptimizationViews.GetGraph() is called.
* OptimizationViews has member INetworkPtr.
* OptimizationViews has GetINetwork() method.
* Unit test added to OptimizationViewsTests.cpp.
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ifc1e53f1c34d786502279631942f0472f401038e
Diffstat (limited to 'src/armnn/test/SubgraphViewTests.cpp')
| -rw-r--r-- | src/armnn/test/SubgraphViewTests.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnn/test/SubgraphViewTests.cpp b/src/armnn/test/SubgraphViewTests.cpp index a1a57c1b56..639e832b87 100644 --- a/src/armnn/test/SubgraphViewTests.cpp +++ b/src/armnn/test/SubgraphViewTests.cpp @@ -200,7 +200,9 @@ TEST_CASE("SingleInputSingleOutputAddPrecompiledLayerSubstituteSubgraph1") // Construct dummy pre-compiled layer INetworkPtr network = INetwork::Create(); - IConnectableLayer* preCompiledLayer = network->AddPrecompiledLayer(preCompiledDescriptor, compiledBlobPtr, backend); + IConnectableLayer* preCompiledLayer = network->AddPrecompiledLayer(preCompiledDescriptor, + compiledBlobPtr, + backend); // Substitute sub-graph with pre-compiled layer graph.SubstituteSubgraph(*subgraph, preCompiledLayer); @@ -242,7 +244,9 @@ TEST_CASE("SingleInputSingleOutputAddPrecompiledLayerSubstituteSubgraph2") // Construct dummy pre-compiled layer INetworkPtr network = INetwork::Create(); - IConnectableLayer* preCompiledLayer = network->AddPrecompiledLayer(preCompiledDescriptor, compiledBlobPtr, backend); + IConnectableLayer* preCompiledLayer = network->AddPrecompiledLayer(preCompiledDescriptor, + compiledBlobPtr, + backend); SubgraphView substituteSubgraph(preCompiledLayer); // Substitute sub-graph with pre-compiled layer |