aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefStackWorkload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/RefStackWorkload.hpp')
-rw-r--r--src/backends/reference/workloads/RefStackWorkload.hpp5
1 files changed, 4 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/RefStackWorkload.hpp b/src/backends/reference/workloads/RefStackWorkload.hpp
index ceb27d9f60..4276339a8f 100644
--- a/src/backends/reference/workloads/RefStackWorkload.hpp
+++ b/src/backends/reference/workloads/RefStackWorkload.hpp
@@ -16,7 +16,10 @@ class RefStackWorkload : public BaseWorkload<StackQueueDescriptor>
public:
explicit RefStackWorkload(const StackQueueDescriptor& descriptor,
const WorkloadInfo& info);
- virtual void Execute() const override;
+ void Execute() const override;
+ void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override;
+private:
+ void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const;
};
} // namespace armnn