From 27dee45706c666d8d1fe66cafe5cea81e68b28bf Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Fri, 7 May 2021 13:46:23 +0100 Subject: IVGCVSW-5987 Print Reduce Descriptor on dot graph Signed-off-by: Teresa Charlin Change-Id: I565a510de0dabdc25d518698174cd61d796239f4 --- include/armnn/TypesUtils.hpp | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'include/armnn') diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 1012fcfa22..1e1926ab76 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -123,6 +123,17 @@ constexpr char const* GetPaddingMethodAsCString(PaddingMethod method) } } +constexpr char const* GetReduceOperationAsCString(ReduceOperation reduce_operation) +{ + switch (reduce_operation) + { + case ReduceOperation::Sum: return "Sum"; + case ReduceOperation::Max: return "Max"; + case ReduceOperation::Mean: return "Mean"; + case ReduceOperation::Min: return "Min"; + default: return "Unknown"; + } +} constexpr unsigned int GetDataTypeSize(DataType dataType) { switch (dataType) -- cgit v1.2.1