aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/RefBackend.cpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2020-07-10 17:50:53 +0100
committerJim Flynn <jim.flynn@arm.com>2020-07-13 09:17:43 +0000
commit549cb7ad5fe0686a5216557ef7ee21b520bf7936 (patch)
tree34c1ff0c8933cb72129d8b8d30d6e814c272841c /src/backends/reference/RefBackend.cpp
parent87d0bda9b49d9df4455f1887027e5ead2527c27e (diff)
downloadarmnn-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/backends/reference/RefBackend.cpp')
-rw-r--r--src/backends/reference/RefBackend.cpp1
1 files changed, 1 insertions, 0 deletions
diff --git a/src/backends/reference/RefBackend.cpp b/src/backends/reference/RefBackend.cpp
index 590fde3e16..e93b317dce 100644
--- a/src/backends/reference/RefBackend.cpp
+++ b/src/backends/reference/RefBackend.cpp
@@ -37,6 +37,7 @@ IBackendInternal::IWorkloadFactoryPtr RefBackend::CreateWorkloadFactory(
auto memoryManager = std::make_shared<RefMemoryManager>();
tensorHandleFactoryRegistry.RegisterMemoryManager(memoryManager);
+ tensorHandleFactoryRegistry.RegisterFactory(std::make_unique<RefTensorHandleFactory>(memoryManager));
return std::make_unique<RefWorkloadFactory>(PolymorphicPointerDowncast<RefMemoryManager>(memoryManager));
}