ArmNN
NotReleased
ClMeanWorkload.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/CLReduceMean.h>
11
12
namespace
armnn
13
{
14
15
arm_compute::Status
ClMeanValidate
(
const
TensorInfo& input,
16
const
TensorInfo& output,
17
const
MeanDescriptor& desc);
18
19
class
ClMeanWorkload
:
public
BaseWorkload
<MeanQueueDescriptor>
20
{
21
public
:
22
ClMeanWorkload
(
const
MeanQueueDescriptor
& descriptor,
const
WorkloadInfo
&
info
);
23
24
void
Execute
()
const override
;
25
26
private
:
27
// Not using CLMeanStdDev, as 4D input tensor support for Mean has been added to a new function called CLReduceMean.
28
mutable
arm_compute::CLReduceMean m_Layer;
29
};
30
31
}
//namespace armnn
armnn::Status
Status
Definition:
Types.hpp:26
armnn::BoostLogSeverityMapping::info
Workload.hpp
armnn::MeanQueueDescriptor
Definition:
WorkloadData.hpp:246
armnn::WorkloadInfo
Definition:
WorkloadInfo.hpp:16
armnn::ClMeanWorkload::ClMeanWorkload
ClMeanWorkload(const MeanQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition:
ClMeanWorkload.cpp:31
armnn
Definition:
BackendHelper.hpp:11
armnn::ClMeanValidate
arm_compute::Status ClMeanValidate(const TensorInfo &input, const TensorInfo &output, const MeanDescriptor &desc)
Definition:
ClMeanWorkload.cpp:17
armnn::ClMeanWorkload
Definition:
ClMeanWorkload.hpp:19
armnn::BaseWorkload
Definition:
Workload.hpp:28
armnn::ClMeanWorkload::Execute
void Execute() const override
Definition:
ClMeanWorkload.cpp:46
src
backends
cl
workloads
ClMeanWorkload.hpp
Generated on Fri Mar 13 2020 16:06:57 for ArmNN by
1.8.13