aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/LoadedNetwork.cpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2020-08-14 11:51:12 +0100
committerJim Flynn <jim.flynn@arm.com>2020-08-16 14:47:00 +0000
commitb8d771ac2e6f847a64b3b203591c5b1c3e198d3a (patch)
tree289b769e179e495e45f2d2c2f9374703be32f9c2 /src/armnn/LoadedNetwork.cpp
parent9e132f57e3fc9d1cd12e3bca2dd3eb82549d7d84 (diff)
downloadarmnn-b8d771ac2e6f847a64b3b203591c5b1c3e198d3a.tar.gz
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 <narumol.prangnawarat@arm.com> Change-Id: Ic2e141e12774bf6d915e77745b6f6d2d83d9b82d
Diffstat (limited to 'src/armnn/LoadedNetwork.cpp')
-rw-r--r--src/armnn/LoadedNetwork.cpp1
1 files changed, 1 insertions, 0 deletions
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<OptimizedNetwork> 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);