• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //
2 // Copyright © 2022 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 "RefBaseWorkload.hpp"
11 #include <armnn/backends/WorkloadData.hpp>
12 
13 namespace armnn
14 {
15 
16 class RefLogicalBinaryWorkload : public RefBaseWorkload<LogicalBinaryQueueDescriptor>
17 {
18 public:
19     using RefBaseWorkload<LogicalBinaryQueueDescriptor>::m_Data;
20 
21     RefLogicalBinaryWorkload(const LogicalBinaryQueueDescriptor& descriptor, const WorkloadInfo& info);
22     void Execute() const override;
23     void ExecuteAsync(ExecutionData& executionData)  override;
24 
25 private:
26     void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const;
27     using InType  = bool;
28     using OutType = bool;
29 };
30 
31 } // namespace armnn
32