src/backends/cl/workloads/ClFloorFloatWorkload.cpp

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