mirror of
https://github.com/Qortal/Brooklyn.git
synced 2025-02-22 07:05:54 +00:00
28 lines
661 B
C++
28 lines
661 B
C++
|
//
|
||
|
|
// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
//
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "TensorHandle.hpp"
|
||
|
|
#include "Workload.hpp"
|
||
|
|
|
||
|
|
#include <utility>
|
||
|
|
|
||
|
|
namespace armnn
|
||
|
|
{
|
||
|
|
|
||
|
|
class CopyMemGenericWorkload : public BaseWorkload<MemCopyQueueDescriptor>
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
CopyMemGenericWorkload(const MemCopyQueueDescriptor& descriptor, const WorkloadInfo& info);
|
||
|
|
void Execute() const override;
|
||
|
|
void ExecuteAsync(WorkingMemDescriptor& descriptor) override;
|
||
|
|
|
||
|
|
private:
|
||
|
|
using TensorHandlePair = std::pair<const ITensorHandle*, ITensorHandle*>;
|
||
|
|
std::vector<TensorHandlePair> m_TensorHandlePairs;
|
||
|
|
};
|
||
|
|
|
||
|
|
} //namespace armnn
|