1 // 2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "OutputHandler.hpp" 7 8 #include <armnn/backends/ITensorHandle.hpp> 9 #include <backendsCommon/WorkloadDataCollector.hpp> 10 #include <backendsCommon/WorkloadFactory.hpp> 11 12 namespace armnn 13 { 14 SetTensorInfo(const TensorInfo & tensorInfo)15void OutputHandler::SetTensorInfo(const TensorInfo& tensorInfo) 16 { 17 m_TensorInfo = tensorInfo; 18 m_bTensorInfoSet = true; 19 } 20 CreateTensorHandles(const IWorkloadFactory & factory,const bool IsMemoryManaged)21void OutputHandler::CreateTensorHandles(const IWorkloadFactory& factory, const bool IsMemoryManaged) 22 { 23 ARMNN_NO_DEPRECATE_WARN_BEGIN 24 m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo, IsMemoryManaged); 25 ARMNN_NO_DEPRECATE_WARN_END 26 } 27 CreateTensorHandles(const ITensorHandleFactory & factory,const bool IsMemoryManaged)28void OutputHandler::CreateTensorHandles(const ITensorHandleFactory& factory, const bool IsMemoryManaged) 29 { 30 m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo, IsMemoryManaged); 31 } 32 CollectWorkloadOutputs(WorkloadDataCollector & dataCollector) const33void OutputHandler::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const 34 { 35 dataCollector.Push(m_TensorHandle.get(), m_TensorInfo); 36 } 37 38 } // namespace armnn 39