From 97520101bb98a30315dd6e31198b08ba050b58c0 Mon Sep 17 00:00:00 2001
From: Ferran Balaguer <ferran.balaguer@arm.com>
Date: Wed, 14 Aug 2019 12:11:27 +0100
Subject: IVGCVSW-3636 Fix Graph and WorkloaData to support backend Import
 functionality

Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
Change-Id: I634aa3b1d609ca33b196fd68ce7fb7881be73e6e
---
 src/armnn/Graph.cpp | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

(limited to 'src/armnn/Graph.cpp')

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
                     {
-- 
cgit v1.2.1