ArmNN
 24.08
ArgMinMax.hpp
Go to the documentation of this file.
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 
Descriptors.hpp
armnn::ArgMinMaxFunction
ArgMinMaxFunction
Definition: Types.hpp:103
armnn::ArgMinMax
void ArgMinMax(Decoder< float > &in, OUT *out, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, ArgMinMaxFunction function, int axis)
Definition: ArgMinMax.cpp:17
Tensor.hpp
Decoders.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6