aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/armnn/layers/StridedSliceLayer.cpp6
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;
}