src/backends/cl/workloads/ClResizeWorkload.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5
   6 #include "ClResizeWorkload.hpp"
   7
   8 #include "ClWorkloadUtils.hpp"
   9
  10 #include <aclCommon/ArmComputeUtils.hpp>
  11 #include <aclCommon/ArmComputeTensorUtils.hpp>
  12
  13 #include <backendsCommon/CpuTensorHandle.hpp>
  14
  15 #include <cl/ClLayerSupport.hpp>
  16 #include <cl/ClTensorHandle.hpp>
  17
  18 using namespace armnn::armcomputetensorutils;
  19
  20 namespace armnn
  21 {
  22
  23 arm_compute::Status ClResizeWorkloadValidate(const TensorInfo& input,
  24                                              const TensorInfo& output,
  25                                              const ResizeDescriptor& descriptor)
  26 {
  27     arm_compute::TensorInfo aclInputInfo  = BuildArmComputeTensorInfo(input);
  28     arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output);
  29
  30     arm_compute::DataLayout aclDataLayout = ConvertDataLayout(descriptor.m_DataLayout);
  31     aclInputInfo.set_data_layout(aclDataLayout);
  32     aclOutputInfo.set_data_layout(aclDataLayout);
  33
  34     arm_compute::InterpolationPolicy aclInterpolationPolicy =
  35         ConvertResizeMethodToAclInterpolationPolicy(descriptor.m_Method);
  36
  37     return arm_compute::CLScale::validate(&aclInputInfo,
  38                                           &aclOutputInfo,
  39                                           aclInterpolationPolicy,
  40                                           arm_compute::BorderMode::REPLICATE,
  41                                           arm_compute::PixelValue(0.f),
  42                                           arm_compute::SamplingPolicy::TOP_LEFT);
  43 }
  44
  45 ClResizeWorkload::ClResizeWorkload(const ResizeQueueDescriptor& descriptor, const WorkloadInfo& info) :
  46     BaseWorkload<ResizeQueueDescriptor>(descriptor, info)
  47 {
  48     m_Data.ValidateInputsOutputs("ClResizeWorkload", 1, 1);
  49
  50     arm_compute::ICLTensor& input  = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
  51     arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
  52
  53     arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
  54     input.info()->set_data_layout(aclDataLayout);
  55     output.info()->set_data_layout(aclDataLayout);
  56
  57     arm_compute::InterpolationPolicy aclInterpolationPolicy =
  58         ConvertResizeMethodToAclInterpolationPolicy(descriptor.m_Parameters.m_Method);
  59
  60     m_ResizeLayer.configure(&input,
  61                             &output,
  62                             aclInterpolationPolicy,
  63                             arm_compute::BorderMode::REPLICATE,
  64                             arm_compute::PixelValue(0.f),
  65                             arm_compute::SamplingPolicy::TOP_LEFT);
  66 };
  67
  68 void ClResizeWorkload::Execute() const
  69 {
  70     ARMNN_SCOPED_PROFILING_EVENT_CL("ClResizeWorkload_Execute");
  71     RunClFunction(m_ResizeLayer, CHECK_LOCATION());
  72 }
  73
  74 } //namespace armnn