1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "NeonSubtractionWorkload.hpp"
7
8 #include "NeonWorkloadUtils.hpp"
9
10 #include <aclCommon/ArmComputeTensorUtils.hpp>
11 #include <aclCommon/ArmComputeUtils.hpp>
12
13 #include <armnn/utility/PolymorphicDowncast.hpp>
14
15 #include <armnn/backends/TensorHandle.hpp>
16
17 #include <arm_compute/runtime/NEON/functions/NEArithmeticSubtraction.h>
18
19 namespace armnn
20 {
21
NeonSubtractionWorkloadValidate(const TensorInfo & input0,const TensorInfo & input1,const TensorInfo & output,const ActivationDescriptor * activationDescriptor)22 arm_compute::Status NeonSubtractionWorkloadValidate(const TensorInfo& input0,
23 const TensorInfo& input1,
24 const TensorInfo& output,
25 const ActivationDescriptor* activationDescriptor)
26 {
27 const arm_compute::TensorInfo aclInput0 = armcomputetensorutils::BuildArmComputeTensorInfo(input0);
28 const arm_compute::TensorInfo aclInput1 = armcomputetensorutils::BuildArmComputeTensorInfo(input1);
29 const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output);
30
31 const arm_compute::ActivationLayerInfo activationInfo = ConvertActivationDescriptorToAclActivationLayerInfo(
32 activationDescriptor);
33
34 return arm_compute::NEArithmeticSubtraction::validate(&aclInput0,
35 &aclInput1,
36 &aclOutput,
37 arm_compute::ConvertPolicy::SATURATE,
38 activationInfo);
39 }
40
NeonSubtractionWorkload(const SubtractionQueueDescriptor & descriptor,const WorkloadInfo & info)41 NeonSubtractionWorkload::NeonSubtractionWorkload(const SubtractionQueueDescriptor& descriptor,
42 const WorkloadInfo& info)
43 : NeonBaseWorkload<SubtractionQueueDescriptor>(descriptor, info)
44 {
45 m_Data.ValidateInputsOutputs("NeonSubtractionWorkload", 2, 1);
46
47 arm_compute::ITensor& input1 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
48 arm_compute::ITensor& input2 = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[1])->GetTensor();
49 arm_compute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
50
51 const arm_compute::ActivationLayerInfo activationInfo = ConvertAdditionalInfoToAclActivationLayerInfo(descriptor);
52
53 auto layer = std::make_unique<arm_compute::NEArithmeticSubtraction>();
54 layer->configure(&input1, &input2, &output, arm_compute::ConvertPolicy::SATURATE, activationInfo);
55 m_SubLayer.reset(layer.release());
56 }
57
Execute() const58 void NeonSubtractionWorkload::Execute() const
59 {
60 ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID("NeonSubtractionWorkload_Execute", this->GetGuid());
61 m_SubLayer->run();
62 }
63
64 } //namespace armnn
65