ArgMinMax.hpp

Go to the documentation of this file.

//
// Copyright © 2019 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

#pragma once

#include "armnn/Tensor.hpp"
#include "armnn/Descriptors.hpp"

#include "Decoders.hpp"

namespace armnn
{

template <typename OUT>
void ArgMinMax(Decoder<float>& in, OUT *out, const TensorInfo& inputTensorInfo,
               const TensorInfo& outputTensorInfo, ArgMinMaxFunction function, int axis);

} //namespace armnn