src/backends/cl/workloads/ClReshapeWorkload.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5
   6 #include "ClReshapeWorkload.hpp"
   7 #include <cl/ClTensorHandle.hpp>
   8 #include <backendsCommon/CpuTensorHandle.hpp>
   9
  10 #include "ClWorkloadUtils.hpp"
  11
  12 namespace armnn
  13 {
  14
  15 ClReshapeWorkload::ClReshapeWorkload(const ReshapeQueueDescriptor& descriptor, const WorkloadInfo& info)
  16     : BaseWorkload<ReshapeQueueDescriptor>(descriptor, info)
  17 {
  18     m_Data.ValidateInputsOutputs("ClReshapeWorkload", 1, 1);
  19
  20     arm_compute::ICLTensor& input  = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetTensor();
  21     arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTensor();
  22
  23     m_Layer.configure(&input, &output);
  24 }
  25
  26 void ClReshapeWorkload::Execute() const
  27 {
  28     ARMNN_SCOPED_PROFILING_EVENT_CL("ClReshapeWorkload_Execute");
  29     RunClFunction(m_Layer, CHECK_LOCATION());
  30 }
  31
  32 } //namespace armnn