src/backends/backendsCommon/MemCopyWorkload.hpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5 #pragma once
   6
   7 #include "CpuTensorHandleFwd.hpp"
   8 #include "Workload.hpp"
   9 #include "WorkloadUtils.hpp"
  10
  11 #include <utility>
  12
  13 namespace armnn
  14 {
  15
  16 class CopyMemGenericWorkload : public BaseWorkload<MemCopyQueueDescriptor>
  17 {
  18 public:
  19     CopyMemGenericWorkload(const MemCopyQueueDescriptor& descriptor, const WorkloadInfo& info);
  20     void Execute() const override;
  21
  22 private:
  23     using TensorHandlePair = std::pair<const ITensorHandle*, ITensorHandle*>;
  24     std::vector<TensorHandlePair> m_TensorHandlePairs;
  25 };
  26
  27 } //namespace armnn