diff options
Diffstat (limited to 'src/armnn/layers/StridedSliceLayer.cpp')
-rw-r--r-- | src/armnn/layers/StridedSliceLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/StridedSliceLayer.cpp b/src/armnn/layers/StridedSliceLayer.cpp index a3dca25656..4ad9b89dd1 100644 --- a/src/armnn/layers/StridedSliceLayer.cpp +++ b/src/armnn/layers/StridedSliceLayer.cpp @@ -65,7 +65,7 @@ std::vector<TensorShape> StridedSliceLayer::InferOutputShapes( int newSize = stride > 0 ? ((stop - start) + stride - 1) / stride : ((start - stop) - stride - 1) / -stride; - newSize = std::min(0, newSize); + newSize = std::max(0, newSize); outputShape.push_back(boost::numeric_cast<unsigned int>(newSize)); } |