diff options
Diffstat (limited to 'src/armnnUtils/DataLayoutIndexed.cpp')
-rw-r--r-- | src/armnnUtils/DataLayoutIndexed.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnnUtils/DataLayoutIndexed.cpp b/src/armnnUtils/DataLayoutIndexed.cpp index c1c98fc0fd..01505a0a31 100644 --- a/src/armnnUtils/DataLayoutIndexed.cpp +++ b/src/armnnUtils/DataLayoutIndexed.cpp @@ -31,6 +31,12 @@ DataLayoutIndexed::DataLayoutIndexed(armnn::DataLayout dataLayout) m_WidthIndex = 3; m_ChannelsIndex = 4; break; + case armnn::DataLayout::NCDHW: + m_ChannelsIndex = 1; + m_DepthIndex = 2; + m_HeightIndex = 3; + m_WidthIndex = 4; + break; default: throw armnn::InvalidArgumentException("Unknown DataLayout value: " + std::to_string(static_cast<int>(dataLayout))); |