ArmNN
 25.11
Loading...
Searching...
No Matches
ClGatherWorkload.hpp
Go to the documentation of this file.
1//
2// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
8#include "ClBaseWorkload.hpp"
9
10#include <arm_compute/runtime/CL/functions/CLGather.h>
11
12namespace armnn
13{
14arm_compute::Status ClGatherWorkloadValidate(const TensorInfo& input,
15 const TensorInfo& indices,
16 const TensorInfo& output,
17 const GatherDescriptor& descriptor);
18
19class ClGatherWorkload : public ClBaseWorkload<GatherQueueDescriptor>
20{
21public:
23 const WorkloadInfo& info,
24 const arm_compute::CLCompileContext& clCompileContext);
25 void Execute() const override;
26
27private:
28 mutable arm_compute::CLGather m_Layer;
29};
30
31} // namespace armnn
ClBaseWorkload(const GatherQueueDescriptor &descriptor, const WorkloadInfo &info)
ClGatherWorkload(const GatherQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
void Execute() const override
Copyright (c) 2021 ARM Limited and Contributors.
arm_compute::Status ClGatherWorkloadValidate(const TensorInfo &input, const TensorInfo &indices, const TensorInfo &output, const GatherDescriptor &descriptor)
A GatherDescriptor for the GatherLayer.
Contains information about TensorInfos of a layer.