From ef6f300928e54cbe6bbfb9dfeb26c2db56ee5dc9 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Mon, 17 Aug 2020 17:02:12 +0100 Subject: IVGCVSW-5114 Enable memory import in TfLiteYoloV3Big App * Enable memory import in TfLiteYoloV3Big App * Add isMemoryManaged flag to Concat and Splitter layers Signed-off-by: Narumol Prangnawarat Change-Id: I7e00f5da2a016c09d480b744fb17ea5611af8365 --- src/armnn/layers/SplitterLayer.cpp | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) (limited to 'src/armnn/layers/SplitterLayer.cpp') diff --git a/src/armnn/layers/SplitterLayer.cpp b/src/armnn/layers/SplitterLayer.cpp index 75fc5378db..e5c9903e2f 100644 --- a/src/armnn/layers/SplitterLayer.cpp +++ b/src/armnn/layers/SplitterLayer.cpp @@ -33,7 +33,9 @@ std::unique_ptr SplitterLayer::CreateWorkload(const IWorkloadFactory& } template -void SplitterLayer::CreateTensors(const TensorHandleFactoryRegistry& registry, const FactoryType& factory) +void SplitterLayer::CreateTensors(const TensorHandleFactoryRegistry& registry, + const FactoryType& factory, + bool isMemoryManaged) { //If sub tensors are supported than all the "splitter" need to do is to //set the outputs to be appropriate sub tensors of the input. @@ -166,28 +168,27 @@ void SplitterLayer::CreateTensors(const TensorHandleFactoryRegistry& registry, c { for (unsigned int i = 0; i < m_Param.GetNumViews(); ++i) { - m_OutputHandlers[i].CreateTensorHandles(factory); + m_OutputHandlers[i].CreateTensorHandles(factory, isMemoryManaged); } } } void SplitterLayer::CreateTensorHandles(const TensorHandleFactoryRegistry& registry, const IWorkloadFactory& workloadFactory, - const bool IsMemoryManaged) + const bool isMemoryManaged) { - IgnoreUnused(IsMemoryManaged); OutputSlot& slot = GetOutputSlot(0); ITensorHandleFactory::FactoryId factoryId = slot.GetTensorHandleFactoryId(); if (factoryId == ITensorHandleFactory::LegacyFactoryId) { - CreateTensors(registry, workloadFactory); + CreateTensors(registry, workloadFactory, isMemoryManaged); } else { ITensorHandleFactory* handleFactory = registry.GetFactory(factoryId); ARMNN_ASSERT(handleFactory); - CreateTensors(registry, *handleFactory); + CreateTensors(registry, *handleFactory, isMemoryManaged); } } -- cgit v1.2.1