commit ec5f7d13582d1e477dc3473223b503388092a352
author Teresa Charlin <teresa.charlinreyes@arm.com> Fri Oct 22 17:15:00 2021 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Wed Oct 27 09:32:11 2021 +0000
tree 37c4d6299f0726a853e28f6297d4eb222cda2c68
parent e1fdd2866b0f403b5e80994890d62c2c038c16c9

IVGCVSW-6170 Add CpuAcc Conv3d Workload

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I3a5bfef5a0085d172fd3689e67f25af909ace2ee

src/backends/aclCommon/ArmComputeTensorUtils.cpp

diff --git a/src/backends/aclCommon/ArmComputeTensorUtils.cpp b/src/backends/aclCommon/ArmComputeTensorUtils.cpp
index 0630931..62f3263 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 @@
 
         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)) + "]");
     }