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

1 files changed, 6 insertions, 0 deletions

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;
         }