diff options
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); } } } |