1 // 2 // Copyright © 2020 Samsung Electronics Co Ltd and Contributors. All rights reserved. 3 // Copyright © 2021-2022 Arm Ltd and Contributors. All rights reserved. 4 // SPDX-License-Identifier: MIT 5 // 6 7 #pragma once 8 9 #include "RefBaseWorkload.hpp" 10 #include <armnn/backends/WorkloadData.hpp> 11 12 namespace armnn 13 { 14 15 class RefReduceWorkload : public RefBaseWorkload<ReduceQueueDescriptor> 16 { 17 public: 18 explicit RefReduceWorkload(const ReduceQueueDescriptor& descriptor, 19 const WorkloadInfo& info); 20 21 void Execute() const override; 22 void ExecuteAsync(ExecutionData& executionData) override; 23 private: 24 void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const; 25 }; 26 27 } //namespace armnn 28