Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / ea8ce7040476da46e145705b0b08e9449144a3b2 / . / src / backends / neon / workloads / NeonResizeWorkload.cpp
blob: fc378cae2ff705b2b2aabe80f437844de1a4e635 [file] [log] [blame]
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]1//
2// Copyright © 2017 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#include "NeonResizeWorkload.hpp"
7
8#include "NeonWorkloadUtils.hpp"
9
10#include <aclCommon/ArmComputeUtils.hpp>
11#include <aclCommon/ArmComputeTensorUtils.hpp>
Jan Eilersbb446e52020-04-02 13:56:54 +0100[diff] [blame]12#include <armnn/utility/PolymorphicDowncast.hpp>
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]13#include <backendsCommon/CpuTensorHandle.hpp>
Jan Eilersbb446e52020-04-02 13:56:54 +0100[diff] [blame]14
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]15#include <neon/NeonTensorHandle.hpp>
16
17using namespace armnn::armcomputetensorutils;
18
19namespace armnn
20{
21
22arm_compute::Status NeonResizeWorkloadValidate(const TensorInfo& input,
23 const TensorInfo& output,
24 const ResizeDescriptor& descriptor)
25{
26 arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input);
27 arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output);
28
29 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(descriptor.m_DataLayout);
30 aclInputInfo.set_data_layout(aclDataLayout);
31 aclOutputInfo.set_data_layout(aclDataLayout);
32
33 arm_compute::InterpolationPolicy aclInterpolationPolicy =
34 ConvertResizeMethodToAclInterpolationPolicy(descriptor.m_Method);
35
David Monahan9b14bfc2020-06-30 15:57:56 +0100[diff] [blame]36 arm_compute::SamplingPolicy samplingPolicy = descriptor.m_HalfPixelCenters ? arm_compute::SamplingPolicy::CENTER :
37 arm_compute::SamplingPolicy::TOP_LEFT;
38
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]39 return arm_compute::NEScale::validate(&aclInputInfo,
40 &aclOutputInfo,
David Monahan9b14bfc2020-06-30 15:57:56 +0100[diff] [blame]41 arm_compute::ScaleKernelInfo(aclInterpolationPolicy,
42 arm_compute::BorderMode::REPLICATE,
43 arm_compute::PixelValue(0.f),
44 samplingPolicy,
45 true,
46 descriptor.m_AlignCorners));
47
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]48}
49
50NeonResizeWorkload::NeonResizeWorkload(const ResizeQueueDescriptor& descriptor,
51 const WorkloadInfo& info)
52 : BaseWorkload<ResizeQueueDescriptor>(descriptor, info)
53{
54 m_Data.ValidateInputsOutputs("NeonResizeWorkload", 1, 1);
55
Jan Eilersbb446e52020-04-02 13:56:54 +0100[diff] [blame]56 arm_compute::ITensor& input = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
57 arm_compute::ITensor& output = PolymorphicDowncast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]58
59 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
60 input.info()->set_data_layout(aclDataLayout);
61 output.info()->set_data_layout(aclDataLayout);
62
63 arm_compute::InterpolationPolicy aclInterpolationPolicy =
64 ConvertResizeMethodToAclInterpolationPolicy(descriptor.m_Parameters.m_Method);
65
David Monahan9b14bfc2020-06-30 15:57:56 +0100[diff] [blame]66 arm_compute::SamplingPolicy samplingPolicy = descriptor.m_Parameters.m_HalfPixelCenters
67 ? arm_compute::SamplingPolicy::CENTER
68 : arm_compute::SamplingPolicy::TOP_LEFT;
69
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]70 m_ResizeLayer.configure(&input,
71 &output,
David Monahan9b14bfc2020-06-30 15:57:56 +0100[diff] [blame]72 arm_compute::ScaleKernelInfo(aclInterpolationPolicy,
73 arm_compute::BorderMode::REPLICATE,
74 arm_compute::PixelValue(0.f),
75 samplingPolicy,
76 true,
77 descriptor.m_Parameters.m_AlignCorners));
Ellen Norris-Thompson37e68682019-07-15 14:23:30 +0100[diff] [blame]78};
79
80void NeonResizeWorkload::Execute() const
81{
82 ARMNN_SCOPED_PROFILING_EVENT_NEON("NeonResizeWorkload_Execute");
83 m_ResizeLayer.run();
84}
85
86} //namespace armnn