1 // 2 // Copyright © 2019 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include "armnn/Tensor.hpp" 9 #include "armnn/Descriptors.hpp" 10 11 #include "Decoders.hpp" 12 13 namespace armnn 14 { 15 16 template <typename OUT> 17 void ArgMinMax(Decoder<float>& in, OUT *out, const TensorInfo& inputTensorInfo, 18 const TensorInfo& outputTensorInfo, ArgMinMaxFunction function, int axis); 19 20 } //namespace armnn 21 22