xref: /aosp_15_r20/external/armnn/src/backends/tosaCommon/operatorMappings/Pooling2DOperator.hpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
1 //
2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "TosaOperatorUtils.hpp"
9 
10 #include <Layer.hpp>
11 
12 #include <tosa_serialization_handler.h>
13 
14 using namespace armnn;
15 using namespace tosa;
16 
17 TosaSerializationBasicBlock* ConvertPooling2DToTosaOperator(const Layer* layer,
18                                                             const std::vector<const TensorInfo*>& inputs,
19                                                             const std::vector<const TensorInfo*>& outputs,
20                                                             const Pooling2dDescriptor* poolDescriptor);
21