diff options
Diffstat (limited to 'src/backends/reference/workloads/RefConstantWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefConstantWorkload.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/RefConstantWorkload.hpp b/src/backends/reference/workloads/RefConstantWorkload.hpp index ada488a7b2..9af5903329 100644 --- a/src/backends/reference/workloads/RefConstantWorkload.hpp +++ b/src/backends/reference/workloads/RefConstantWorkload.hpp @@ -19,8 +19,10 @@ class RefConstantWorkload : public BaseWorkload<ConstantQueueDescriptor> public: RefConstantWorkload(const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info); - void PostAllocationConfigure() override; - virtual void Execute() const override; + void Execute() const override; + void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override; +private: + void Execute(std::vector<ITensorHandle*> outputs) const; }; } //namespace armnn |