diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/TypesUtils.hpp | 40 |
1 files changed, 29 insertions, 11 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 065b6839fc..8157d4f043 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -77,16 +77,6 @@ constexpr char const* GetPoolingAlgorithmAsCString(PoolingAlgorithm pooling) } } -constexpr char const* GetResizeMethodAsCString(ResizeMethod resizeMethod) -{ - switch (resizeMethod) - { - case ResizeMethod::Bilinear: return "Bilinear"; - case ResizeMethod::NearestNeighbor: return "NearestNeighbor"; - default: return "Unknown"; - } -} - constexpr char const* GetOutputShapeRoundingAsCString(OutputShapeRounding rounding) { switch (rounding) @@ -97,7 +87,6 @@ constexpr char const* GetOutputShapeRoundingAsCString(OutputShapeRounding roundi } } - constexpr char const* GetPaddingMethodAsCString(PaddingMethod method) { switch (method) @@ -185,6 +174,35 @@ 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"; + } +} + +constexpr const char* GetResizeMethodAsCString(ResizeMethod method) +{ + switch (method) + { + case ResizeMethod::Bilinear: return "Bilinear"; + case ResizeMethod::NearestNeighbor: return "NearestNeighbour"; + default: return "Unknown"; + } +} template<typename T> struct IsHalfType |