aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/MockBackend.cpp
diff options
context:
space:
mode:
authorCathal Corbett <cathal.corbett@arm.com>2021-12-15 17:12:59 +0000
committerCathal Corbett <cathal.corbett@arm.com>2021-12-23 13:21:22 +0000
commitcbfd718464b8ac41f0338ae6565d8213d24c0a2a (patch)
treef26da835108a0ed52ac0ffc8f7ebec64827b5033 /src/backends/backendsCommon/test/MockBackend.cpp
parent81edc6217f76953c0be4c47f3d005cf48772ccb7 (diff)
downloadarmnn-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/backends/backendsCommon/test/MockBackend.cpp')
-rw-r--r--src/backends/backendsCommon/test/MockBackend.cpp12
1 files changed, 8 insertions, 4 deletions
diff --git a/src/backends/backendsCommon/test/MockBackend.cpp b/src/backends/backendsCommon/test/MockBackend.cpp
index 4bdb8ce786..a34cfe98fe 100644
--- a/src/backends/backendsCommon/test/MockBackend.cpp
+++ b/src/backends/backendsCommon/test/MockBackend.cpp
@@ -187,12 +187,16 @@ OptimizationViews MockBackend::OptimizeSubgraphView(const SubgraphView& subgraph
{
ARMNN_ASSERT(supportedSubgraph != nullptr);
- PreCompiledLayer* preCompiledLayer =
- optimizationViews.GetGraph().AddLayer<PreCompiledLayer>(
+ CompiledBlobPtr blobPtr;
+ BackendId backend = MockBackendId();
+
+ IConnectableLayer* preCompiledLayer =
+ optimizationViews.GetINetwork()->AddPrecompiledLayer(
PreCompiledDescriptor(supportedSubgraph->GetNumInputSlots(),
supportedSubgraph->GetNumOutputSlots()),
- "pre-compiled");
- preCompiledLayer->SetBackendId(MockBackendId());
+ blobPtr,
+ backend,
+ nullptr);
SubgraphView substitutionSubgraph(*supportedSubgraph);
SubgraphView replacementSubgraph(preCompiledLayer);