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 : BaseWorkload<LogicalBinaryQueueDescriptor>(descriptor, info)
35 {
36 m_Data.ValidateInputsOutputs("NeonLogicalOrWorkload", 2, 1);
37
38 arm_compute::ITensor& input0 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
39 arm_compute::ITensor& input1 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor();
40 arm_compute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
41
42 m_LogicalOrLayer.configure(&input0, &input1, &output);
43 }
44
Execute() const45 void NeonLogicalOrWorkload::Execute() const
46 {
47 ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonLogicalOrWorkload_Execute");
48 m_LogicalOrLayer.run();
49 }
50
51 } // namespace armnn
52