20.11
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
Tensor.hpp
armnn::ArgMinMax
void ArgMinMax(Decoder< float > &in, OUT *out, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, ArgMinMaxFunction function, int axis)
Definition:
ArgMinMax.cpp:16
armnn
Copyright (c) 2020 ARM Limited.
Definition:
00_introduction.dox:25
Decoders.hpp
Descriptors.hpp
armnn::ArgMinMaxFunction
ArgMinMaxFunction
Definition:
Types.hpp:72
src
backends
reference
workloads
ArgMinMax.hpp
Generated on Wed Nov 25 2020 09:59:14 for ArmNN by
1.8.13