blob: 03e458c409136734f38ec756b1b606b2ecb0ac41 [file] [log] [blame]
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +01001//
Mike Kelly7cbe7812023-07-25 17:37:33 +01002// Copyright © 2019-2023 Arm Ltd and Contributors. All rights reserved.
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +01003// SPDX-License-Identifier: MIT
4//
5
6#include "NeonAbsWorkload.hpp"
7
8#include "NeonWorkloadUtils.hpp"
9
10#include <aclCommon/ArmComputeTensorHandle.hpp>
11#include <aclCommon/ArmComputeTensorUtils.hpp>
Jan Eilersbb446e52020-04-02 13:56:54 +010012#include <armnn/utility/PolymorphicDowncast.hpp>
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +010013
14namespace armnn
15{
16
17arm_compute::Status NeonAbsWorkloadValidate(const TensorInfo& input, const TensorInfo& output)
18{
19 const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input);
20 const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output);
21
22 return arm_compute::NEAbsLayer::validate(&aclInput, &aclOutput);
23}
24
25NeonAbsWorkload::NeonAbsWorkload(const AbsQueueDescriptor& descriptor, const WorkloadInfo& info)
Teresa Charlin588cbdf2022-01-19 15:55:37 +000026 : NeonBaseWorkload<AbsQueueDescriptor>(descriptor, info)
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +010027{
28 m_Data.ValidateInputsOutputs("NeonAbsWorkload", 1, 1);
29
Jan Eilersbb446e52020-04-02 13:56:54 +010030 arm_compute::ITensor& input = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
31 arm_compute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +010032
33 m_AbsLayer.configure(&input, &output);
34}
35
36void NeonAbsWorkload::Execute() const
37{
Mike Kelly7cbe7812023-07-25 17:37:33 +010038 ARMNN_SCOPED_PROFILING_EVENT_NEON_NAME_GUID("NeonAbsWorkload_Execute");
Aron Virginas-Tar914e4db2019-09-09 13:36:45 +010039 m_AbsLayer.run();
40}
41
42} // namespace armnn