diff options
Diffstat (limited to 'include/armnn/TypesUtils.hpp')
-rw-r--r-- | include/armnn/TypesUtils.hpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 8157d4f043..790f57a432 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -66,6 +66,19 @@ constexpr char const* GetComparisonOperationAsCString(ComparisonOperation operat } } +constexpr char const* GetUnaryOperationAsCString(UnaryOperation operation) +{ + switch (operation) + { + case UnaryOperation::Abs: return "Abs"; + case UnaryOperation::Exp: return "Exp"; + case UnaryOperation::Sqrt: return "Sqrt"; + case UnaryOperation::Rsqrt: return "Rsqrt"; + case UnaryOperation::Neg: return "Neg"; + default: return "Unknown"; + } +} + constexpr char const* GetPoolingAlgorithmAsCString(PoolingAlgorithm pooling) { switch (pooling) |