diff options
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/layers/StridedSliceLayer.cpp | 6 |
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; } |