diff options
Diffstat (limited to 'include/armnn/TypesUtils.hpp')
-rw-r--r-- | include/armnn/TypesUtils.hpp | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 32967c6b42..20bc28c8ec 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -183,6 +183,25 @@ constexpr const char* GetDataLayoutName(DataLayout dataLayout) } } +constexpr const char* GetNormalizationAlgorithmChannelAsCString(NormalizationAlgorithmChannel channel) +{ + switch (channel) + { + case NormalizationAlgorithmChannel::Across: return "Across"; + case NormalizationAlgorithmChannel::Within: return "Within"; + default: return "Unknown"; + } +} + +constexpr const char* GetNormalizationAlgorithmMethodAsCString(NormalizationAlgorithmMethod method) +{ + switch (method) + { + case NormalizationAlgorithmMethod::LocalBrightness: return "LocalBrightness"; + case NormalizationAlgorithmMethod::LocalContrast: return "LocalContrast"; + default: return "Unknown"; + } +} template<typename T> struct IsHalfType |