From 4dc64a69ba383ece509d442598617445a3b4847f Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Mon, 16 Sep 2019 17:00:22 +0100 Subject: IVGCVSW-3694 Add ArgMinMax implementation for Ref * Add ArgMinMax implementation * Add utility function to get number of elements between axis * Add utility function to get unsigned axis * Unit tests for ArgMinMax function Signed-off-by: Narumol Prangnawarat Change-Id: I7bc3d610dda9526190187eb87394a8ed7a4b5cdd --- src/armnnUtils/TensorUtils.hpp | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'src/armnnUtils/TensorUtils.hpp') diff --git a/src/armnnUtils/TensorUtils.hpp b/src/armnnUtils/TensorUtils.hpp index 03b1c8a2df..2b1f6a24f3 100644 --- a/src/armnnUtils/TensorUtils.hpp +++ b/src/armnnUtils/TensorUtils.hpp @@ -26,4 +26,10 @@ std::pair FindMinMax(armnn::ITensorHandle* tensorHandle); armnn::TensorShape ExpandDims(const armnn::TensorShape& tensorShape, int axis); +unsigned int GetNumElementsBetween(const armnn::TensorShape& shape, + unsigned int firstAxisInclusive, + unsigned int lastAxisExclusive); + +unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis); + } // namespace armnnUtils -- cgit v1.2.1