IVGCVSW-4531 StridedSlice Failures on HAL Drivers
* Check if the output is large enough to hold the slice in the driver
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Iaef16167e891a41186de030324bb45b8bbca4d80
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp
index 1538133..8313d04 100644
--- a/ConversionUtils.hpp
+++ b/ConversionUtils.hpp
@@ -3599,6 +3599,26 @@
return false;
}
+ // Check if slice can fit in a inferred output
+ armnn::TensorShape inputShape = inputInfo.GetShape();
+ for (unsigned int i = 0; i < inputShape.GetNumDimensions(); i++)
+ {
+ int stride = descriptor.m_Stride[i];
+ int start = descriptor.GetStartForAxis(inputShape, i);
+ int stop = descriptor.GetStopForAxis(inputShape, i, start);
+
+ if (descriptor.m_ShrinkAxisMask & (1 << i))
+ {
+ // If the difference between the start point and the end point of the slice on an axis being shrunk
+ // is greater than 1 then throw an error as the output will not be large enough to hold the slice
+ if (((descriptor.m_Begin[i] - descriptor.m_End[i]) > 1)
+ || ((descriptor.m_Begin[i] - descriptor.m_End[i]) < -1))
+ {
+ return Fail("%s: StridedSlice: Output will not be large enough to hold the slice", __func__);
+ }
+ }
+ }
+
armnn::IConnectableLayer* const layer = data.m_Network->AddStridedSliceLayer(descriptor);
assert(layer != nullptr);
input.Connect(layer->GetInputSlot(0));