xref: /aosp_15_r20/external/armnn/src/backends/neon/workloads/NeonFillWorkload.hpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
1 //
2 // Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <armnn/backends/WorkloadData.hpp>
9 #include "NeonBaseWorkload.hpp"
10 #include <arm_compute/runtime/IFunction.h>
11 #include <arm_compute/core/Error.h>
12 
13 namespace armnn {
14 
15 class NeonFillWorkload : public NeonBaseWorkload<FillQueueDescriptor>
16 {
17 public:
18     NeonFillWorkload(const FillQueueDescriptor& descriptor, const WorkloadInfo& info);
19     void Execute() const override;
20 
21 private:
22     std::unique_ptr<arm_compute::IFunction> m_Layer;
23 };
24 
25 } //namespace armnn
26