From b7c1831f95a5ecdde0fff068d4054a066309cec6 Mon Sep 17 00:00:00 2001
From: Ryan OShea <Ryan.OShea2@arm.com>
Date: Fri, 3 Jul 2020 11:40:12 +0100
Subject: IVGCVSW-4920 Invalid Negative Stride fix

 * Add check for negative stride with ShrinkAxisMask

Signed-off-by: Ryan OShea <Ryan.OShea2@arm.com>
Change-Id: I69ecd2915d94278d6313d8279361c4dc675adb74
---
 src/armnn/layers/StridedSliceLayer.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/armnn/layers/StridedSliceLayer.cpp b/src/armnn/layers/StridedSliceLayer.cpp
index 2609b94d91..ae4fab0efd 100644
--- a/src/armnn/layers/StridedSliceLayer.cpp
+++ b/src/armnn/layers/StridedSliceLayer.cpp
@@ -65,6 +65,12 @@ std::vector<TensorShape> StridedSliceLayer::InferOutputShapes(
                 throw LayerValidationException(
                     "StridedSlice: Attempting to take a larger slice than can fit in inferred output");
             }
+
+            if (stride < 0)
+            {
+                throw LayerValidationException(
+                    "StridedSlice: Stride can not be negative with Shrink Axis Mask set.");
+            }
             continue;
         }
 
-- 
cgit v1.2.1