diff options
Diffstat (limited to 'src/armnn/Graph.cpp')
| -rw-r--r-- | src/armnn/Graph.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/Graph.cpp b/src/armnn/Graph.cpp index f0ef0e18f7..5e2acd55a3 100644 --- a/src/armnn/Graph.cpp +++ b/src/armnn/Graph.cpp @@ -335,7 +335,6 @@ void Graph::AddCompatibilityLayers(std::map<BackendId, std::unique_ptr<IBackendI auto backend = backendIt->second.get(); auto tensorHandleFactoryIds = backend->GetHandleFactoryPreferences(); bool found = false; - boost::ignore_unused(found); for (auto preference : tensorHandleFactoryIds) { @@ -344,10 +343,12 @@ void Graph::AddCompatibilityLayers(std::map<BackendId, std::unique_ptr<IBackendI { auto srcPref = srcOutputSlot.GetTensorHandleFactoryId(); auto srcFactory = registry.GetFactory(srcPref); + if (srcFactory) { bool canExportImport = - (factory->GetImportFlags() & srcFactory->GetExportFlags()) != 0; + (factory->GetImportFlags() & srcFactory->GetExportFlags()) != 0; + if (factory->SupportsMapUnmap() || canExportImport) { compOutputSlot.SetTensorHandleFactory(preference); |