1 // 2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include "RefBaseWorkload.hpp" 9 #include <armnn/backends/WorkloadData.hpp> 10 11 #include <armnn/Types.hpp> 12 13 namespace armnn 14 { 15 16 // Base class template providing an implementation of the Constant layer common to all data types. 17 class RefConstantWorkload : public RefBaseWorkload<ConstantQueueDescriptor> 18 { 19 public: 20 RefConstantWorkload(const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info); 21 22 void Execute() const override; 23 void ExecuteAsync(ExecutionData& executionData) override; 24 private: 25 void Execute(std::vector<ITensorHandle*> outputs) const; 26 }; 27 28 } //namespace armnn 29