IVGCVSW-3623 Implement NeonTensorHandle::Import

Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I7213788725fd4e4cf1176998604e999d0b7ed6cc

1 files changed, 15 insertions, 2 deletions

diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp
index 1000eceda0..7ee4e612e0 100644
--- a/src/armnn/LoadedNetwork.cpp
+++ b/src/armnn/LoadedNetwork.cpp
@@ -120,8 +120,21 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr<OptimizedNetwork> net,
 
     for (auto&& layer : order)
     {
-        auto& workloadFacory = GetWorkloadFactory(*layer);
-        layer->CreateTensorHandles(m_TensorHandleFactoryRegistry, workloadFacory);
+        auto& workloadFactory = GetWorkloadFactory(*layer);
+
+        switch (layer->GetType())
+        {
+        case LayerType::Input:
+            {
+                // If IsImportEnabled is true then we need to set IsMemoryManaged to false when creating TensorHandles
+                layer->CreateTensorHandles(m_TensorHandleFactoryRegistry, workloadFactory, !m_IsImportEnabled);
+                break;
+            }
+        default:
+            {
+                layer->CreateTensorHandles(m_TensorHandleFactoryRegistry, workloadFactory);
+            }
+        }
     }
 
     //Then create workloads.