ArmNN
NotReleased
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
void
ArgMinMax
(Decoder<float>& in, int32_t* out,
const
TensorInfo& inputTensorInfo,
17
const
TensorInfo& outputTensorInfo,
ArgMinMaxFunction
function
,
int
axis);
18
19
}
//namespace armnn
20
armnn::ArgMinMax
void ArgMinMax(Decoder< float > &in, int32_t *out, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, ArgMinMaxFunction function, int axis)
Definition:
ArgMinMax.cpp:15
Descriptors.hpp
Decoders.hpp
armnn
Definition:
BackendHelper.hpp:11
armnn::ArgMinMaxFunction
ArgMinMaxFunction
Definition:
Types.hpp:68
Tensor.hpp
src
backends
reference
workloads
ArgMinMax.hpp
Generated on Fri Mar 13 2020 16:06:58 for ArmNN by
1.8.13