diff options
author | David Monahan <david.monahan@arm.com> | 2019-08-20 11:25:29 +0100 |
---|---|---|
committer | David Monahan <david.monahan@arm.com> | 2019-09-24 10:50:30 +0000 |
commit | 3fb7e105ae62cbfb3ebf1edebb90e2b6672b22aa (patch) | |
tree | 47793a736e1fb53d51b4c0fd755f4e24f7f93d98 /src/armnn/Layer.cpp | |
parent | 93667b1d7c361df68bdb1d733f17aba3ba34e046 (diff) | |
download | armnn-3fb7e105ae62cbfb3ebf1edebb90e2b6672b22aa.tar.gz |
IVGCVSW-3623 Implement NeonTensorHandle::Import
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I7213788725fd4e4cf1176998604e999d0b7ed6cc
Diffstat (limited to 'src/armnn/Layer.cpp')
-rw-r--r-- | src/armnn/Layer.cpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/armnn/Layer.cpp b/src/armnn/Layer.cpp index 1e384336c9..dbeda22ca0 100644 --- a/src/armnn/Layer.cpp +++ b/src/armnn/Layer.cpp @@ -244,7 +244,9 @@ void Layer::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector, const G } } -void Layer::CreateTensorHandles(const TensorHandleFactoryRegistry& registry, const IWorkloadFactory& workloadFactory) +void Layer::CreateTensorHandles(const TensorHandleFactoryRegistry& registry, + const IWorkloadFactory& workloadFactory, + const bool IsMemoryManaged) { for (unsigned int idx=0; idx < GetNumOutputSlots(); idx++) { @@ -255,14 +257,13 @@ void Layer::CreateTensorHandles(const TensorHandleFactoryRegistry& registry, con OutputHandler& handler = GetOutputHandler(idx); if (factoryId == ITensorHandleFactory::LegacyFactoryId) { - handler.CreateTensorHandles(workloadFactory); + handler.CreateTensorHandles(workloadFactory, IsMemoryManaged); } else { ITensorHandleFactory* handleFactory = registry.GetFactory(factoryId); BOOST_ASSERT(handleFactory); - - handler.CreateTensorHandles(*handleFactory); + handler.CreateTensorHandles(*handleFactory, IsMemoryManaged); } } } |