From b8d771ac2e6f847a64b3b203591c5b1c3e198d3a Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Fri, 14 Aug 2020 11:51:12 +0100 Subject: IVGCVSW-5012 Enable zero copy for Neon * Allow memory import if padding is not required in Neon * AddMockImportBackend for fallback tests * Refactor GraphUtils * Memory import unit tests * Fallback unit tests Signed-off-by: Narumol Prangnawarat Change-Id: Ic2e141e12774bf6d915e77745b6f6d2d83d9b82d --- src/armnn/LoadedNetwork.cpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/armnn/LoadedNetwork.cpp') diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index 593539d3ee..4a293b92d9 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -168,6 +168,7 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr net, switch (layer->GetType()) { case LayerType::Input: + case LayerType::MemImport: { // If IsImportEnabled is true then we need to set IsMemoryManaged to false when creating TensorHandles layer->CreateTensorHandles(m_TensorHandleFactoryRegistry, workloadFactory, !m_IsImportEnabled); -- cgit v1.2.1