diff options
author | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-08-14 12:11:27 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-08-14 13:09:48 +0000 |
commit | 97520101bb98a30315dd6e31198b08ba050b58c0 (patch) | |
tree | 7ddde38e3feb3858208063afabbb380722ba66e1 /src/armnn | |
parent | 4f77ac2685a4130b07a4d28420929d883b251571 (diff) | |
download | armnn-97520101bb98a30315dd6e31198b08ba050b58c0.tar.gz |
IVGCVSW-3636 Fix Graph and WorkloaData to support backend Import functionality
Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
Change-Id: I634aa3b1d609ca33b196fd68ce7fb7881be73e6e
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/Graph.cpp | 19 |
1 files changed, 13 insertions, 6 deletions
diff --git a/src/armnn/Graph.cpp b/src/armnn/Graph.cpp index 6212c49eba..f0ef0e18f7 100644 --- a/src/armnn/Graph.cpp +++ b/src/armnn/Graph.cpp @@ -344,17 +344,24 @@ void Graph::AddCompatibilityLayers(std::map<BackendId, std::unique_ptr<IBackendI { auto srcPref = srcOutputSlot.GetTensorHandleFactoryId(); auto srcFactory = registry.GetFactory(srcPref); - bool canExportImport = (factory->GetImportFlags() & srcFactory->GetExportFlags()) != 0; - if (factory->SupportsMapUnmap() || canExportImport) + if (srcFactory) { - compOutputSlot.SetTensorHandleFactory(preference); - found = true; - break; + bool canExportImport = + (factory->GetImportFlags() & srcFactory->GetExportFlags()) != 0; + if (factory->SupportsMapUnmap() || canExportImport) + { + compOutputSlot.SetTensorHandleFactory(preference); + found = true; + break; + } } } } - BOOST_ASSERT_MSG(found, "Could not find a valid TensorHandle for compatibilty layer"); + if (!found) + { + compOutputSlot.SetTensorHandleFactory(ITensorHandleFactory::LegacyFactoryId); + } } else { |