• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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