• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <backendsCommon/Workload.hpp>
9 
10 #include <arm_compute/core/Error.h>
11 #include <arm_compute/runtime/IFunction.h>
12 
13 #include <functional>
14 
15 namespace armnn
16 {
17 
18 arm_compute::Status NeonDequantizeWorkloadValidate(const TensorInfo& input,
19                                                    const TensorInfo& output);
20 
21 class NeonDequantizeWorkload : public BaseWorkload<DequantizeQueueDescriptor>
22 {
23 public:
24     NeonDequantizeWorkload(const DequantizeQueueDescriptor& descriptor, const WorkloadInfo& info);
25 
26     void Execute() const override;
27 
28 private:
29     mutable std::unique_ptr<arm_compute::IFunction> m_Layer;
30 };
31 
32 } //namespace armnn
33