diff options
Diffstat (limited to 'include/armnn/TypesUtils.hpp')
-rw-r--r-- | include/armnn/TypesUtils.hpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 7eacc00a93..8c4ceb8d4f 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -135,6 +135,16 @@ constexpr const char* GetDataTypeName(DataType dataType) } } +constexpr const char* GetDataLayoutName(DataLayout dataLayout) +{ + switch (dataLayout) + { + case DataLayout::NCHW: return "NCHW"; + case DataLayout::NHWC: return "NHWC"; + default: return "Unknown"; + } +} + template<typename T> struct IsHalfType |