src/backends/reference/workloads/ArgMinMax.hpp

   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 void ArgMinMax(Decoder<float>& in, int32_t* out, const TensorInfo& inputTensorInfo,
  17                const TensorInfo& outputTensorInfo, ArgMinMaxFunction function, int axis);
  18
  19 } //namespace armnn
  20