src/backends/backendsCommon/WorkloadDataCollector.hpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5 #pragma once
   6
   7 #include <armnn/Tensor.hpp>
   8
   9 #include <vector>
  10
  11 namespace armnn
  12 {
  13 class ITensorHandle;
  14
  15 class WorkloadDataCollector
  16 {
  17 public:
  18     WorkloadDataCollector(std::vector<ITensorHandle*>& handles, std::vector<TensorInfo>& infos)
  19         : m_Handles(handles)
  20         , m_Infos(infos)
  21     {
  22     }
  23
  24     void Push(ITensorHandle* handle, const TensorInfo& info)
  25     {
  26         m_Handles.push_back(handle);
  27         m_Infos.push_back(info);
  28     }
  29
  30 private:
  31     std::vector<ITensorHandle*>& m_Handles;
  32     std::vector<TensorInfo>& m_Infos;
  33 };
  34
  35
  36 } //namespace armnn