diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-07-10 17:50:53 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-07-13 09:17:43 +0000 |
commit | 549cb7ad5fe0686a5216557ef7ee21b520bf7936 (patch) | |
tree | 34c1ff0c8933cb72129d8b8d30d6e814c272841c /src/armnn/LoadedNetwork.cpp | |
parent | 87d0bda9b49d9df4455f1887027e5ead2527c27e (diff) | |
download | armnn-549cb7ad5fe0686a5216557ef7ee21b520bf7936.tar.gz |
IVGCVSW-5078 Fix segmentation fault on Yolo V3 Big App
* Register TensorHandleFactories when create Workload Factory
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I6217c6f00d27b67e42b8043cbaf72d9db5789bfb
Diffstat (limited to 'src/armnn/LoadedNetwork.cpp')
-rw-r--r-- | src/armnn/LoadedNetwork.cpp | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index b35dfd1107..0dd75aa649 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -121,6 +121,7 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr<OptimizedNetwork> net, m_OptimizedNetwork(std::move(net)), m_IsImportEnabled(networkProperties.m_ImportEnabled), m_IsExportEnabled(networkProperties.m_ExportEnabled), + m_TensorHandleFactoryRegistry(), m_ProfilingService(profilingService) { // Create a profiler and register it for the current thread. @@ -144,8 +145,6 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr<OptimizedNetwork> net, if (backend->SupportsTensorAllocatorAPI()) { - backend->RegisterTensorHandleFactories(m_TensorHandleFactoryRegistry); - auto workloadFactory = backend->CreateWorkloadFactory(m_TensorHandleFactoryRegistry); m_WorkloadFactories.emplace( std::make_pair(backendId, std::make_pair(std::move(workloadFactory), nullptr))); |