src/backends/backendsCommon/OutputHandler.cpp

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