ArmNN
 22.11
ArgMinMax.hpp File Reference
#include "armnn/Tensor.hpp"
#include "armnn/Descriptors.hpp"
#include "Decoders.hpp"

Go to the source code of this file.

Namespaces

 armnn
 Copyright (c) 2021 ARM Limited and Contributors.
 

Functions

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