ArmNN
 25.11
Loading...
Searching...
No Matches
ClMeanWorkload.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
8#include "ClBaseWorkload.hpp"
9
10#include <arm_compute/runtime/CL/functions/CLReduceMean.h>
11
12namespace armnn
13{
14
15arm_compute::Status ClMeanValidate(const TensorInfo& input,
16 const TensorInfo& output,
17 const MeanDescriptor& descriptor);
18
19class ClMeanWorkload : public ClBaseWorkload<MeanQueueDescriptor>
20{
21public:
22 ClMeanWorkload(const MeanQueueDescriptor& descriptor,
23 const WorkloadInfo& info,
24 const arm_compute::CLCompileContext& clCompileContext);
25
26 void Execute() const override;
27
28private:
29 // Not using CLMeanStdDev, as 4D input tensor support for Mean has been added to a new function called CLReduceMean.
30 mutable arm_compute::CLReduceMean m_Layer;
31};
32
33} //namespace armnn
ClBaseWorkload(const MeanQueueDescriptor &descriptor, const WorkloadInfo &info)
ClMeanWorkload(const MeanQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
void Execute() const override
Copyright (c) 2021 ARM Limited and Contributors.
arm_compute::Status ClMeanValidate(const TensorInfo &input, const TensorInfo &output, const MeanDescriptor &descriptor)
A MeanDescriptor for the MeanLayer.
Contains information about TensorInfos of a layer.