diff options
Diffstat (limited to 'src/backends/reference/workloads/RefElementwiseUnaryWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefElementwiseUnaryWorkload.hpp | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/RefElementwiseUnaryWorkload.hpp b/src/backends/reference/workloads/RefElementwiseUnaryWorkload.hpp index efb2865ebd..d05347bbe5 100644 --- a/src/backends/reference/workloads/RefElementwiseUnaryWorkload.hpp +++ b/src/backends/reference/workloads/RefElementwiseUnaryWorkload.hpp @@ -19,15 +19,13 @@ public: using BaseWorkload<ElementwiseUnaryQueueDescriptor>::m_Data; RefElementwiseUnaryWorkload(const ElementwiseUnaryQueueDescriptor& descriptor, const WorkloadInfo& info); - void PostAllocationConfigure() override; void Execute() const override; + void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override; private: + void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const; using InType = float; using OutType = float; - - std::unique_ptr<Decoder<InType>> m_Input; - std::unique_ptr<Encoder<OutType>> m_Output; }; } // namespace armnn |