commit a0d2844d05dd9ae15733d426b04ab651457604ce
author Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> Wed Nov 21 16:48:17 2018 +0000
committer nattapat.chaimanowong <nattapat.chaimanowong@arm.com> Wed Nov 21 17:13:30 2018 +0000
tree 7ae0787d7be49ec2b1bb754ff35e13e8b71c73d2
parent 3d93bc47f42b339d82cfcf56a90c9264dd46d70a

IVGCVSW-2086 Update StridedSliceLayer and StridedSliceDescriptor

Change-Id: Ifa88a879dd239f60ab27330d6b73859393828ef0

src/backends/backendsCommon/WorkloadData.cpp

diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index d5e3638..af57fee 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -946,6 +946,12 @@
     const TensorInfo& input = workloadInfo.m_InputTensorInfos[0];
     const uint32_t rank = input.GetNumDimensions();
 
+    if (rank > 4)
+    {
+        throw InvalidArgumentException(
+            "StridedSliceLayer: Input tensors with rank greater than 4 are not supported");
+    }
+
     // Begin, End & Stride length must be of rank(input0)
     if (m_Parameters.m_Begin.size() != rank)
     {
@@ -975,4 +981,4 @@
     }
 }
 
-} //namespace armnn
\ No newline at end of file
+} //namespace armnn