1 // 2 // Copyright © 2021 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include "BaseIterator.hpp" 9 #include "Decoders.hpp" 10 #include "Encoders.hpp" 11 12 #include <armnn/Tensor.hpp> 13 14 namespace armnn 15 { 16 17 void Reduce(const TensorInfo& inputInfo, 18 const TensorInfo& outputInfo, 19 Decoder<float>& input, 20 Encoder<float>& output, 21 const std::vector<uint32_t> axis, 22 const ReduceOperation reduceOperation); 23 24 } //namespace armnn 25