10 #include <boost/numeric/conversion/cast.hpp> 23 const unsigned int axisSize = inputTensorInfo.
GetShape()[uAxis];
28 for (
unsigned int outer = 0; outer < outerElements; ++outer) {
29 for (
unsigned int inner = 0; inner < innerElements; ++inner) {
30 in[outer * axisSize * innerElements + inner];
31 auto tmpValue = in.
Get();
32 unsigned int tmpIndex = 0;
33 for (
unsigned int i = 1; i < axisSize; ++i) {
34 in[(outer * axisSize * innerElements) + (i * innerElements) + inner];
35 const auto& value = in.
Get();
const TensorShape & GetShape() const
void ArgMinMax(Decoder< float > &in, int32_t *out, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, ArgMinMaxFunction function, int axis)
unsigned int GetNumElementsBetween(const armnn::TensorShape &shape, unsigned int firstAxisInclusive, unsigned int lastAxisExclusive)
Copyright (c) 2020 ARM Limited.
void IgnoreUnused(Ts &&...)
virtual IType Get() const =0
unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis)
std::enable_if_t< std::is_unsigned< Source >::value &&std::is_unsigned< Dest >::value, Dest > numeric_cast(Source source)
unsigned int GetNumDimensions() const