diff options
Diffstat (limited to 'src/backends/reference/workloads/RefQuantizeWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefQuantizeWorkload.hpp | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/RefQuantizeWorkload.hpp b/src/backends/reference/workloads/RefQuantizeWorkload.hpp index 9ae107607b..48116e7b39 100644 --- a/src/backends/reference/workloads/RefQuantizeWorkload.hpp +++ b/src/backends/reference/workloads/RefQuantizeWorkload.hpp @@ -16,13 +16,11 @@ class RefQuantizeWorkload : public BaseWorkload<QuantizeQueueDescriptor> { public: RefQuantizeWorkload(const QuantizeQueueDescriptor& descriptor, const WorkloadInfo &info); - void PostAllocationConfigure() override; void Execute() const override; + void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override; private: - - std::unique_ptr<Decoder<float>> m_InputDecoder; - std::unique_ptr<Encoder<float>> m_OutputEncoder; + void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const; size_t m_NumElements; }; |