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