1 //
2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "RefLogicalUnaryWorkload.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
RefLogicalUnaryWorkload(const ElementwiseUnaryQueueDescriptor & desc,const WorkloadInfo & info)20 RefLogicalUnaryWorkload::RefLogicalUnaryWorkload(const ElementwiseUnaryQueueDescriptor& desc,
21 const WorkloadInfo& info)
22 : RefBaseWorkload<ElementwiseUnaryQueueDescriptor>(desc, info)
23 {}
24
Execute() const25 void RefLogicalUnaryWorkload::Execute() const
26 {
27 Execute(m_Data.m_Inputs, m_Data.m_Outputs);
28 }
29
ExecuteAsync(ExecutionData & executionData)30 void RefLogicalUnaryWorkload::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 RefLogicalUnaryWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const
37 {
38 ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefLogicalUnaryWorkload_Execute");
39
40 const TensorInfo& inputInfo = GetTensorInfo(inputs[0]);
41 const TensorInfo& outputInfo = GetTensorInfo(outputs[0]);
42
43 const TensorShape& inShape = inputInfo.GetShape();
44 const TensorShape& outShape = outputInfo.GetShape();
45
46 std::unique_ptr<Decoder<InType>> input = MakeDecoder<InType>(inputInfo, inputs[0]->Map());
47 std::unique_ptr<Encoder<OutType>> output = MakeEncoder<OutType>(outputInfo, outputs[0]->Map());
48
49 using NotFunction = LogicalUnaryFunction<std::logical_not<bool>>;
50
51 switch (m_Data.m_Parameters.m_Operation)
52 {
53 case UnaryOperation::LogicalNot:
54 {
55 NotFunction(inShape, outShape, *input, *output);
56 break;
57 }
58 default:
59 {
60 throw InvalidArgumentException(std::string("Unsupported Logical Unary operation") +
61 GetUnaryOperationAsCString(m_Data.m_Parameters.m_Operation), CHECK_LOCATION());
62 }
63 }
64 }
65
66 } // namespace armnn
67