1 //
2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "RefLogicalBinaryWorkload.hpp"
7
8 #include "Decoders.hpp"
9 #include "ElementwiseFunction.hpp"
10 #include "Encoders.hpp"
11 #include "RefWorkloadUtils.hpp"
12
13 #include <Profiling.hpp>
14
15 #include <armnn/TypesUtils.hpp>
16
17 namespace armnn
18 {
19
RefLogicalBinaryWorkload(const LogicalBinaryQueueDescriptor & desc,const WorkloadInfo & info)20 RefLogicalBinaryWorkload::RefLogicalBinaryWorkload(const LogicalBinaryQueueDescriptor& desc,
21 const WorkloadInfo& info)
22 : RefBaseWorkload<LogicalBinaryQueueDescriptor>(desc, info)
23 {}
24
Execute() const25 void RefLogicalBinaryWorkload::Execute() const
26 {
27 Execute(m_Data.m_Inputs, m_Data.m_Outputs);
28 }
29
ExecuteAsync(ExecutionData & executionData)30 void RefLogicalBinaryWorkload::ExecuteAsync(ExecutionData& executionData)
31 {
32 WorkingMemDescriptor* workingMemDescriptor = static_cast<WorkingMemDescriptor*>(executionData.m_Data);
33 Execute(workingMemDescriptor->m_Inputs, workingMemDescriptor->m_Outputs);
34 }
35
Execute(std::vector<ITensorHandle * > inputs,std::vector<ITensorHandle * > outputs) const36 void RefLogicalBinaryWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const
37 {
38 ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefLogicalBinaryWorkload_Execute");
39
40 const TensorInfo& inputInfo0 = GetTensorInfo(inputs[0]);
41 const TensorInfo& inputInfo1 = GetTensorInfo(inputs[1]);
42 const TensorInfo& outputInfo = GetTensorInfo(outputs[0]);
43
44 const TensorShape& inShape0 = inputInfo0.GetShape();
45 const TensorShape& inShape1 = inputInfo1.GetShape();
46 const TensorShape& outShape = outputInfo.GetShape();
47
48 std::unique_ptr<Decoder<InType>> input0 = MakeDecoder<InType>(inputInfo0, inputs[0]->Map());
49 std::unique_ptr<Decoder<InType>> input1 = MakeDecoder<InType>(inputInfo1, inputs[1]->Map());
50 std::unique_ptr<Encoder<OutType>> output = MakeEncoder<OutType>(outputInfo, outputs[0]->Map());
51
52 using AndFunction = LogicalBinaryFunction<std::logical_and<bool>>;
53 using OrFunction = LogicalBinaryFunction<std::logical_or<bool>>;
54
55 switch (m_Data.m_Parameters.m_Operation)
56 {
57 case LogicalBinaryOperation::LogicalAnd:
58 {
59 AndFunction(inShape0, inShape1, outShape, *input0, *input1, *output);
60 break;
61 }
62 case LogicalBinaryOperation::LogicalOr:
63 {
64 OrFunction(inShape0, inShape1, outShape, *input0, *input1, *output);
65 break;
66 }
67 default:
68 {
69 throw InvalidArgumentException(std::string("Unsupported Logical Binary operation") +
70 GetLogicalBinaryOperationAsCString(m_Data.m_Parameters.m_Operation), CHECK_LOCATION());
71 }
72 }
73 }
74
75 } // namespace armnn
76