1 files changed, 7 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index d5e3638a06..af57fee935 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -946,6 +946,12 @@ void StridedSliceQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) con
     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 @@ void StridedSliceQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) con
     }
 }
 
-} //namespace armnn
-\ No newline at end of file
+} //namespace armnn