ArmNN
NotReleased
NeonGreaterWorkload.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 <
neon/workloads/NeonWorkloadUtils.hpp
>
9
10
#include <arm_compute/runtime/NEON/functions/NEElementwiseOperations.h>
11
12
namespace
armnn
13
{
14
15
arm_compute::Status
NeonGreaterWorkloadValidate
(
const
TensorInfo& input0,
16
const
TensorInfo& input1,
17
const
TensorInfo& output);
18
19
template
<DataType T>
20
class
NeonGreaterWorkload
:
public
MultiTypedWorkload
<GreaterQueueDescriptor, T, DataType::Boolean>
21
{
22
public
:
23
using
MultiTypedWorkload<GreaterQueueDescriptor, T, DataType::Boolean>::m_Data
;
24
25
NeonGreaterWorkload
(
const
GreaterQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
26
27
virtual
void
Execute
()
const override
;
28
29
private
:
30
mutable
arm_compute::NEGreater m_GreaterLayer;
31
};
32
33
using
NeonGreaterFloat32Workload
=
NeonGreaterWorkload<DataType::Float32>
;
34
using
NeonGreaterUint8Workload
=
NeonGreaterWorkload<DataType::QAsymmU8>
;
35
36
}
//namespace armnn
armnn::Status
Status
Definition:
Types.hpp:26
armnn::BoostLogSeverityMapping::info
NeonWorkloadUtils.hpp
armnn::WorkloadInfo
Definition:
WorkloadInfo.hpp:16
armnn::GreaterQueueDescriptor
Definition:
WorkloadData.hpp:436
armnn::NeonGreaterWorkload::Execute
virtual void Execute() const override
Definition:
NeonGreaterWorkload.cpp:40
armnn::NeonGreaterWorkload
Definition:
NeonGreaterWorkload.hpp:20
armnn::MultiTypedWorkload
Definition:
Workload.hpp:105
armnn::NeonGreaterWorkloadValidate
arm_compute::Status NeonGreaterWorkloadValidate(const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output)
Definition:
NeonGreaterWorkload.cpp:13
armnn
Definition:
BackendHelper.hpp:11
armnn::NeonGreaterWorkload::NeonGreaterWorkload
NeonGreaterWorkload(const GreaterQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
NeonGreaterWorkload.cpp:27
src
backends
neon
workloads
NeonGreaterWorkload.hpp
Generated on Fri Mar 13 2020 16:06:58 for ArmNN by
1.8.13