diff options
-rw-r--r-- | src/armnn/Network.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 932f9eb49e..0ef14075b6 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -441,7 +441,8 @@ bool RequiresCopy(ITensorHandleFactory::FactoryId src, ITensorHandleFactory* srcFactory = registry.GetFactory(src); ITensorHandleFactory* dstFactory = registry.GetFactory(dst); - if ((srcFactory->GetExportFlags() & dstFactory->GetImportFlags()) != 0) + if (srcFactory && dstFactory && + (srcFactory->GetExportFlags() & dstFactory->GetImportFlags()) != 0) { return false; } |