src/backends/cl/workloads/ClResizeBilinearFloatWorkload.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5
   6 #include "ClResizeBilinearFloatWorkload.hpp"
   7 #include <cl/ClTensorHandle.hpp>
   8 #include <backendsCommon/CpuTensorHandle.hpp>
   9 #include <cl/ClLayerSupport.hpp>
  10 #include <aclCommon/ArmComputeUtils.hpp>
  11 #include <aclCommon/ArmComputeTensorUtils.hpp>
  12
  13 #include "ClWorkloadUtils.hpp"
  14
  15 using namespace armnn::armcomputetensorutils;
  16
  17 namespace armnn
  18 {
  19
  20 ClResizeBilinearFloatWorkload::ClResizeBilinearFloatWorkload(const ResizeBilinearQueueDescriptor& descriptor,
  21                                                              const WorkloadInfo& info)
  22     : FloatWorkload<ResizeBilinearQueueDescriptor>(descriptor, info)
  23 {
  24     m_Data.ValidateInputsOutputs("ClResizeBilinearFloatWorkload", 1, 1);
  25
  26     arm_compute::ICLTensor& input  = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
  27     arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
  28
  29     arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout);
  30     input.info()->set_data_layout(aclDataLayout);
  31     output.info()->set_data_layout(aclDataLayout);
  32
  33     m_ResizeBilinearLayer.configure(&input, &output, arm_compute::InterpolationPolicy::BILINEAR,
  34                                     arm_compute::BorderMode::REPLICATE, arm_compute::PixelValue(0.f),
  35                                     arm_compute::SamplingPolicy::TOP_LEFT);
  36 };
  37
  38 void ClResizeBilinearFloatWorkload::Execute() const
  39 {
  40     ARMNN_SCOPED_PROFILING_EVENT_CL("ClResizeBilinearFloatWorkload_Execute");
  41     RunClFunction(m_ResizeBilinearLayer, CHECK_LOCATION());
  42 }
  43
  44 } //namespace armnn