blob: 4a92e6b6dc9ffb32da4b796df24a35fa2ac0c2e0 [file] [log] [blame]
Teresa Charlin2b030d92020-03-27 16:40:56 +00001//
2// Copyright © 2020 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
8#include <backendsCommon/Workload.hpp>
9
10#include <arm_compute/runtime/CL/functions/CLComparison.h>
11
12namespace armnn
13{
14
15arm_compute::Status ClComparisonWorkloadValidate(const TensorInfo& input0,
16 const TensorInfo& input1,
17 const TensorInfo& output,
18 const ComparisonDescriptor& descriptor);
19
20class ClComparisonWorkload : public BaseWorkload<ComparisonQueueDescriptor>
21{
22public:
Sadik Armagane9444752020-12-02 11:28:58 +000023 ClComparisonWorkload(const ComparisonQueueDescriptor& descriptor,
24 const WorkloadInfo& info,
25 const arm_compute::CLCompileContext& clCompileContext);
Teresa Charlin2b030d92020-03-27 16:40:56 +000026 void Execute() const override;
27
28private:
29 mutable arm_compute::CLComparison m_ComparisonLayer;
30};
31
32} //namespace armnn