diff options
Diffstat (limited to 'src/backends/reference/workloads/RefGatherWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefGatherWorkload.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefGatherWorkload.hpp b/src/backends/reference/workloads/RefGatherWorkload.hpp index 30019a8d4d..1664e1611d 100644 --- a/src/backends/reference/workloads/RefGatherWorkload.hpp +++ b/src/backends/reference/workloads/RefGatherWorkload.hpp @@ -21,6 +21,9 @@ class RefGatherWorkload : public BaseWorkload<GatherQueueDescriptor> public: using BaseWorkload<GatherQueueDescriptor>::BaseWorkload; void Execute() const override; + void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override; +private: + void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const; }; } // namespace armnn |