1 // 2 // Copyright © 2017 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include <armnn/Descriptors.hpp> 9 #include <armnn/Tensor.hpp> 10 11 #include "BaseIterator.hpp" 12 13 namespace armnn 14 { 15 /// Computes the Pooling2d operation. 16 void Pooling2d(Decoder<float>& rInputDecoder, 17 Encoder<float>& rOutputEncoder, 18 const TensorInfo& inputInfo, 19 const TensorInfo& outputInfo, 20 const Pooling2dDescriptor& params); 21 } //namespace armnn 22