diff options
Diffstat (limited to 'src/backends/aclCommon/ArmComputeTensorUtils.cpp')
-rw-r--r-- | src/backends/aclCommon/ArmComputeTensorUtils.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/backends/aclCommon/ArmComputeTensorUtils.cpp b/src/backends/aclCommon/ArmComputeTensorUtils.cpp index 06309319f3..62f3263a0c 100644 --- a/src/backends/aclCommon/ArmComputeTensorUtils.cpp +++ b/src/backends/aclCommon/ArmComputeTensorUtils.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #include <aclCommon/ArmComputeTensorUtils.hpp> @@ -143,6 +143,10 @@ arm_compute::DataLayout ConvertDataLayout(armnn::DataLayout dataLayout) case armnn::DataLayout::NCHW : return arm_compute::DataLayout::NCHW; + case armnn::DataLayout::NDHWC : return arm_compute::DataLayout::NDHWC; + + case armnn::DataLayout::NCDHW : return arm_compute::DataLayout::NCDHW; + default: throw InvalidArgumentException("Unknown armnn::DataLayout: [" + std::to_string(static_cast<int>(dataLayout)) + "]"); } |