diff options
Diffstat (limited to 'src/backends/reference/workloads/RefElementwiseWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefElementwiseWorkload.hpp | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/RefElementwiseWorkload.hpp b/src/backends/reference/workloads/RefElementwiseWorkload.hpp index 03683b1a06..4dc4b5ba5a 100644 --- a/src/backends/reference/workloads/RefElementwiseWorkload.hpp +++ b/src/backends/reference/workloads/RefElementwiseWorkload.hpp @@ -26,13 +26,11 @@ public: using BaseWorkload<ParentDescriptor>::m_Data; RefElementwiseWorkload(const ParentDescriptor& descriptor, const WorkloadInfo& info); - void PostAllocationConfigure() override; void Execute() const override; + void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override; private: - std::unique_ptr<Decoder<InType>> m_Input0; - std::unique_ptr<Decoder<InType>> m_Input1; - std::unique_ptr<Encoder<OutType>> m_Output; + void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const; }; template <typename DataType = float> |