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