// | |
// Copyright © 2017 Arm Ltd. All rights reserved. | |
// SPDX-License-Identifier: MIT | |
// | |
#pragma once | |
#include "CpuTensorHandleFwd.hpp" | |
#include "Workload.hpp" | |
#include "WorkloadUtils.hpp" | |
#include <utility> | |
namespace armnn | |
{ | |
class SyncMemGenericWorkload : public BaseWorkload<MemSyncQueueDescriptor> | |
{ | |
public: | |
SyncMemGenericWorkload(const MemSyncQueueDescriptor& descriptor, const WorkloadInfo& info); | |
void Execute() const override; | |
private: | |
ITensorHandle* m_TensorHandle; | |
}; | |
} //namespace armnn |