1 //
2 // Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "NeonLogicalOrWorkload.hpp"
7
8 #include "NeonWorkloadUtils.hpp"
9
10 #include <aclCommon/ArmComputeTensorHandle.hpp>
11 #include <aclCommon/ArmComputeTensorUtils.hpp>
12
13 #include <armnn/utility/PolymorphicDowncast.hpp>
14
15 namespace armnn
16 {
17
NeonLogicalOrWorkloadValidate(const TensorInfo & input0,const TensorInfo & input1,const TensorInfo & output)18 arm_compute::Status NeonLogicalOrWorkloadValidate(const TensorInfo& input0,
19 const TensorInfo& input1,
20 const TensorInfo& output)
21 {
22 const arm_compute::TensorInfo aclInputInfo0 = BuildArmComputeTensorInfo(input0);
23 const arm_compute::TensorInfo aclInputInfo1 = BuildArmComputeTensorInfo(input1);
24 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output);
25
26 const arm_compute::Status aclStatus = arm_compute::NELogicalOr::validate(&aclInputInfo0,
27 &aclInputInfo1,
28 &aclOutputInfo);
29 return aclStatus;
30 }
31
NeonLogicalOrWorkload(const LogicalBinaryQueueDescriptor & descriptor,const WorkloadInfo & info)32 NeonLogicalOrWorkload::NeonLogicalOrWorkload(const LogicalBinaryQueueDescriptor& descriptor,
33 const WorkloadInfo& info)
34 : NeonBaseWorkload<LogicalBinaryQueueDescriptor>(descriptor, info)
35 {
36 // Report Profiling Details
37 ARMNN_REPORT_PROFILING_WORKLOAD_DESC("NeonLogicalOrWorkload_Construct",
38 descriptor.m_Parameters,
39 info,
40 this->GetGuid());
41
42 m_Data.ValidateInputsOutputs("NeonLogicalOrWorkload", 2, 1);
43
44 arm_compute::ITensor& input0 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
45 arm_compute::ITensor& input1 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor();
46 arm_compute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
47
48 m_LogicalOrLayer.configure(&input0, &input1, &output);
49 }
50
Execute() const51 void NeonLogicalOrWorkload::Execute() const
52 {
53 ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID("NeonLogicalOrWorkload_Execute", this->GetGuid());
54 m_LogicalOrLayer.run();
55 }
56
57 } // namespace armnn
58