From 11bd26133423f5076e74a08752954819de56eb26 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Tue, 13 Aug 2019 10:26:53 +0100 Subject: IVGCVSW-3607 Add CreateWorkloadFactory with TensorHandleFactoryRegistry in IBackendInternal Signed-off-by: Narumol Prangnawarat Change-Id: Id40b825649d0447058273b8bc4f7133a3ac3c96c --- src/armnn/LoadedNetwork.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'src/armnn/LoadedNetwork.cpp') diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index a81528aa65..f5f79f3940 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -99,7 +99,7 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr net) { backend->RegisterTensorHandleFactories(m_TensorHandleFactoryRegistry); - auto workloadFactory = backend->CreateWorkloadFactory(); + auto workloadFactory = backend->CreateWorkloadFactory(m_TensorHandleFactoryRegistry); m_WorkloadFactories.emplace( std::make_pair(backendId, std::make_pair(std::move(workloadFactory), nullptr))); } @@ -491,6 +491,7 @@ void LoadedNetwork::AllocateWorkingMemory() memoryManager->Acquire(); } } + m_TensorHandleFactoryRegistry.AquireMemory(); m_IsWorkingMemAllocated = true; } @@ -510,6 +511,7 @@ void LoadedNetwork::FreeWorkingMemory() memoryManager->Release(); } } + m_TensorHandleFactoryRegistry.ReleaseMemory(); m_IsWorkingMemAllocated = false; } -- cgit v1.2.1