aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--include/armnn/TypesUtils.hpp10
1 files changed, 10 insertions, 0 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp
index cdcbd3ca9d..32967c6b42 100644
--- a/include/armnn/TypesUtils.hpp
+++ b/include/armnn/TypesUtils.hpp
@@ -42,6 +42,16 @@ constexpr char const* GetActivationFunctionAsCString(ActivationFunction activati
}
}
+constexpr char const* GetArgMinMaxFunctionAsCString(ArgMinMaxFunction function)
+{
+ switch (function)
+ {
+ case ArgMinMaxFunction::Max: return "Max";
+ case ArgMinMaxFunction::Min: return "Min";
+ default: return "Unknown";
+ }
+}
+
constexpr char const* GetComparisonOperationAsCString(ComparisonOperation operation)
{
switch (operation)