blob: 0812e33a5229741596abd084911564f9771d4bb9 [file] [log] [blame]
Nattapat Chaimanowongac9e0962018-10-10 17:18:35 +01001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
8#include <backends/Workload.hpp>
9
10#include <arm_compute/runtime/CL/CLFunctions.h>
11
12namespace armnn
13{
14
15arm_compute::Status ClPooling2dWorkloadValidate(const TensorInfo& input,
16 const TensorInfo& output,
17 const Pooling2dDescriptor& descriptor);
18
19class ClPooling2dWorkload : public BaseWorkload<Pooling2dQueueDescriptor>
20{
21public:
22 using BaseWorkload<Pooling2dQueueDescriptor>::m_Data;
23
24 ClPooling2dWorkload(const Pooling2dQueueDescriptor& descriptor,
25 const WorkloadInfo& info);
26
27 void Execute() const override;
28
29private:
30 mutable arm_compute::CLPoolingLayer m_PoolingLayer;
31};
32
33} //namespace armnn