ArmNN
 24.02
ClFloorFloatWorkload.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
9 
10 #include <arm_compute/runtime/CL/functions/CLFloor.h>
11 
12 namespace armnn
13 {
14 
15 arm_compute::Status ClFloorWorkloadValidate(const TensorInfo& input,
16  const TensorInfo& output);
17 
18 class ClFloorFloatWorkload : public FloatWorkload<FloorQueueDescriptor>
19 {
20 public:
22  const WorkloadInfo& info,
23  const arm_compute::CLCompileContext& clCompileContext);
24 
25  void Execute() const override;
26  // Replace input tensor handle with the given TensorHandle
27  void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override;
28 
29  // Replace output tensor handle with the given TensorHandle
30  void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override;
31 private:
32  mutable arm_compute::CLFloor m_Layer;
33  virtual void Reconfigure();
34 };
35 
36 } //namespace armnn
37 
38 
39 
40 
armnn::ClFloorWorkloadValidate
arm_compute::Status ClFloorWorkloadValidate(const TensorInfo &input, const TensorInfo &output)
Definition: ClFloorFloatWorkload.cpp:14
armnn::FloorQueueDescriptor
Definition: WorkloadData.hpp:395
armnn::ITensorHandle
Definition: ITensorHandle.hpp:16
armnn::TypedWorkload
Definition: Workload.hpp:101
armnn::ClFloorFloatWorkload
Definition: ClFloorFloatWorkload.hpp:18
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
Workload.hpp
armnn::ClFloorFloatWorkload::ClFloorFloatWorkload
ClFloorFloatWorkload(const FloorQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Definition: ClFloorFloatWorkload.cpp:23
armnn::BoostLogSeverityMapping::info
@ info
armnn::Status
Status
Definition: Types.hpp:42
armnn::ClFloorFloatWorkload::Execute
void Execute() const override
Definition: ClFloorFloatWorkload.cpp:38
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
armnn::ClFloorFloatWorkload::ReplaceOutputTensorHandle
void ReplaceOutputTensorHandle(ITensorHandle *tensorHandle, unsigned int slot) override
Definition: ClFloorFloatWorkload.cpp:61
armnn::ClFloorFloatWorkload::ReplaceInputTensorHandle
void ReplaceInputTensorHandle(ITensorHandle *tensorHandle, unsigned int slot) override
Definition: ClFloorFloatWorkload.cpp:44