From b2d3ec5b1e938ef34facfdbcff83fc8e845d5f7c Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Tue, 12 Apr 2022 22:07:09 +0100 Subject: IVGCVSW-6856 Add GATHERNd FrontEnd and Ref Implementation * Add front end * Add reference workload * Add unit tests * Add EndToEnd test Signed-off-by: Teresa Charlin Change-Id: I4cebd17b18476df86162e2dda3366c10e80bd2f8 --- include/armnn/backends/WorkloadData.hpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'include/armnn/backends/WorkloadData.hpp') diff --git a/include/armnn/backends/WorkloadData.hpp b/include/armnn/backends/WorkloadData.hpp index 21141583c6..ed89f9638c 100644 --- a/include/armnn/backends/WorkloadData.hpp +++ b/include/armnn/backends/WorkloadData.hpp @@ -527,6 +527,11 @@ struct RsqrtQueueDescriptor : QueueDescriptor void Validate(const WorkloadInfo& workloadInfo) const; }; +struct GatherNdQueueDescriptor : QueueDescriptor +{ + void Validate(const WorkloadInfo& workloadInfo) const; +}; + struct GatherQueueDescriptor : QueueDescriptorWithParameters { void Validate(const WorkloadInfo& workloadInfo) const; -- cgit v1.2.1