From 3fb7e105ae62cbfb3ebf1edebb90e2b6672b22aa Mon Sep 17 00:00:00 2001 From: David Monahan Date: Tue, 20 Aug 2019 11:25:29 +0100 Subject: IVGCVSW-3623 Implement NeonTensorHandle::Import Signed-off-by: David Monahan Change-Id: I7213788725fd4e4cf1176998604e999d0b7ed6cc --- src/backends/backendsCommon/OutputHandler.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/backends/backendsCommon/OutputHandler.cpp') diff --git a/src/backends/backendsCommon/OutputHandler.cpp b/src/backends/backendsCommon/OutputHandler.cpp index 8f4942d8ba..e3a1b276ea 100644 --- a/src/backends/backendsCommon/OutputHandler.cpp +++ b/src/backends/backendsCommon/OutputHandler.cpp @@ -22,14 +22,14 @@ void OutputHandler::SetTensorInfo(const TensorInfo& tensorInfo) m_bTensorInfoSet = true; } -void OutputHandler::CreateTensorHandles(const IWorkloadFactory& factory) +void OutputHandler::CreateTensorHandles(const IWorkloadFactory& factory, const bool IsMemoryManaged) { - m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo); + m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo, IsMemoryManaged); } -void OutputHandler::CreateTensorHandles(const ITensorHandleFactory& factory) +void OutputHandler::CreateTensorHandles(const ITensorHandleFactory& factory, const bool IsMemoryManaged) { - m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo); + m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo, IsMemoryManaged); } void OutputHandler::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const -- cgit v1.2.1