xref: /aosp_15_r20/external/armnn/src/backends/neon/workloads/NeonConstantWorkload.hpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "NeonBaseWorkload.hpp"
9 
10 #include <neon/workloads/NeonWorkloadUtils.hpp>
11 
12 namespace armnn
13 {
14 arm_compute::Status NeonConstantWorkloadValidate(const TensorInfo& output);
15 
16 class NeonConstantWorkload : public NeonBaseWorkload<ConstantQueueDescriptor>
17 {
18 public:
19     NeonConstantWorkload(const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info);
20 
21     virtual void Execute() const override;
22 
23 private:
24     mutable bool m_RanOnce;
25 };
26 
27 } //namespace armnn
28