From b89b05f048a566a8c825f1d223966bc5a6abc3d5 Mon Sep 17 00:00:00 2001
From: narpra01 <narumol.prangnawarat@arm.com>
Date: Wed, 16 Jan 2019 09:53:09 +0000
Subject: IVGCVSW-2508 Add no-op factory implementations and layer for Gather
 operator  * Added GatherQueueDescriptor to WorkloadData  * Added CreateGather
 function in WorkloadFactory.hpp  * Added stub implementation of the
 CreateGreater function in workload factories  * Added GatherLayer stub
 implementation  * Added AddGatherLayer to Network  * Added IsGatherSupported
 to LayerSupportBase

Change-Id: I0408fd54e88a7d4e3d9e1c2811a9323f0da52a04
---
 src/backends/reference/RefWorkloadFactory.cpp | 6 ++++++
 src/backends/reference/RefWorkloadFactory.hpp | 3 +++
 2 files changed, 9 insertions(+)

(limited to 'src/backends/reference')

diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp
index 361a3f1f74..cb7d6ea01a 100644
--- a/src/backends/reference/RefWorkloadFactory.cpp
+++ b/src/backends/reference/RefWorkloadFactory.cpp
@@ -324,4 +324,10 @@ std::unique_ptr<IWorkload> RefWorkloadFactory::CreatePreCompiled(const PreCompil
     return nullptr;
 }
 
+std::unique_ptr<IWorkload> RefWorkloadFactory::CreateGather(const armnn::GatherQueueDescriptor& descriptor,
+                                                            const armnn::WorkloadInfo& info) const
+{
+    return MakeWorkload<NullWorkload, NullWorkload>(descriptor, info);
+}
+
 } // namespace armnn
diff --git a/src/backends/reference/RefWorkloadFactory.hpp b/src/backends/reference/RefWorkloadFactory.hpp
index 432ac72c6e..443af76265 100644
--- a/src/backends/reference/RefWorkloadFactory.hpp
+++ b/src/backends/reference/RefWorkloadFactory.hpp
@@ -171,6 +171,9 @@ public:
     virtual std::unique_ptr<IWorkload> CreatePreCompiled(const PreCompiledQueueDescriptor& descriptor,
                                                          const WorkloadInfo& info) const override;
 
+    virtual std::unique_ptr<IWorkload> CreateGather(const GatherQueueDescriptor& descriptor,
+                                                    const WorkloadInfo& info) const override;
+
 private:
 
     template <typename F32Workload, typename U8Workload, typename QueueDescriptorType>
-- 
cgit v1.2.1