diff options
Diffstat (limited to 'include/armnn/TypesUtils.hpp')
-rw-r--r-- | include/armnn/TypesUtils.hpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index 20fcbbd784..b644daafd8 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -218,9 +218,10 @@ constexpr const char* GetDataLayoutName(DataLayout dataLayout) { switch (dataLayout) { - case DataLayout::NCHW: return "NCHW"; - case DataLayout::NHWC: return "NHWC"; - default: return "Unknown"; + case DataLayout::NCHW: return "NCHW"; + case DataLayout::NHWC: return "NHWC"; + case DataLayout::NDHWC: return "NDHWC"; + default: return "Unknown"; } } |