10 #include <boost/numeric/conversion/cast.hpp> 18 boost::ignore_unused(outputTensorInfo);
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();
42 out[outer * innerElements + inner] = boost::numeric_cast<int32_t>(tmpIndex);
unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis)
unsigned int GetNumDimensions() const
void ArgMinMax(Decoder< float > &in, int32_t *out, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, ArgMinMaxFunction function, int axis)
virtual IType Get() const =0
unsigned int GetNumElementsBetween(const armnn::TensorShape &shape, unsigned int firstAxisInclusive, unsigned int lastAxisExclusive)
const TensorShape & GetShape() const