2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
6 #include "OutputHandler.hpp"
8 #include "ITensorHandle.hpp"
9 #include "WorkloadDataCollector.hpp"
11 #include <backendsCommon/WorkloadFactory.hpp>
13 #include <boost/assert.hpp>
14 #include <boost/log/trivial.hpp>
19 void OutputHandler::SetTensorInfo(const TensorInfo& tensorInfo)
21 m_TensorInfo = tensorInfo;
22 m_bTensorInfoSet = true;
25 void OutputHandler::CreateTensorHandles(const IWorkloadFactory& factory)
27 m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo);
30 void OutputHandler::CreateTensorHandles(const IWorkloadFactory& factory, DataLayout dataLayout)
32 m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo, dataLayout);
35 void OutputHandler::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const
37 dataCollector.Push(m_TensorHandle.get(), m_TensorInfo);