ArmNN
NotReleased
ClGreaterWorkload.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 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
12
namespace
armnn
13
{
14
15
arm_compute::Status
ClGreaterWorkloadValidate
(
const
TensorInfo& input0,
16
const
TensorInfo& input1,
17
const
TensorInfo& output);
18
19
template
<DataType T>
20
class
ClGreaterWorkload
:
public
MultiTypedWorkload
<GreaterQueueDescriptor, T, DataType::Boolean>
21
{
22
public
:
23
ClGreaterWorkload
(
const
GreaterQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
24
void
Execute
()
const override
;
25
26
private
:
27
using
MultiTypedWorkload<GreaterQueueDescriptor, T, DataType::Boolean>::m_Data
;
28
mutable
arm_compute::CLComparison m_GreaterLayer;
29
};
30
31
using
ClGreaterFloat32Workload
=
ClGreaterWorkload<DataType::Float32>
;
32
using
ClGreaterUint8Workload
=
ClGreaterWorkload<DataType::QAsymmU8>
;
33
34
}
//namespace armnn
armnn::Status
Status
Definition:
Types.hpp:26
armnn::BoostLogSeverityMapping::info
armnn::ClGreaterWorkload
Definition:
ClGreaterWorkload.hpp:20
armnn::ClGreaterWorkload::ClGreaterWorkload
ClGreaterWorkload(const GreaterQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
ClGreaterWorkload.cpp:42
Workload.hpp
armnn::WorkloadInfo
Definition:
WorkloadInfo.hpp:16
armnn::GreaterQueueDescriptor
Definition:
WorkloadData.hpp:436
armnn::ClGreaterWorkload::Execute
void Execute() const override
Definition:
ClGreaterWorkload.cpp:56
armnn::MultiTypedWorkload
Definition:
Workload.hpp:105
armnn
Definition:
BackendHelper.hpp:11
armnn::ClGreaterWorkloadValidate
arm_compute::Status ClGreaterWorkloadValidate(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output)
Definition:
ClGreaterWorkload.cpp:24
src
backends
cl
workloads
ClGreaterWorkload.hpp
Generated on Fri Mar 13 2020 16:06:57 for ArmNN by
1.8.13