• 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 "BaseIterator.hpp"
9 
10 #include <backendsCommon/Workload.hpp>
11 #include <backendsCommon/WorkloadData.hpp>
12 
13 namespace armnn
14 {
15 
16 class RefLogicalUnaryWorkload : public BaseWorkload<ElementwiseUnaryQueueDescriptor>
17 {
18 public:
19     using BaseWorkload<ElementwiseUnaryQueueDescriptor>::m_Data;
20 
21     RefLogicalUnaryWorkload(const ElementwiseUnaryQueueDescriptor& descriptor, const WorkloadInfo& info);
22     void PostAllocationConfigure() override;
23     virtual void Execute() const override;
24 
25 private:
26     using InType  = bool;
27     using OutType = bool;
28 
29     std::unique_ptr<Decoder<InType>>  m_Input;
30     std::unique_ptr<Encoder<OutType>> m_Output;
31 };
32 
33 } // namespace armnn
34