IVGCVSW-5220 Fix UnknownCombinationsTest/UnknownDimensionsTest
Signed-off-by: Finn Williams <Finn.Williams@Arm.com>
Change-Id: I4729497e67b711a8f1fbf7c555a26e97fa6c9ab6
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp
index f2f95ac..51727a3 100644
--- a/ConversionUtils.hpp
+++ b/ConversionUtils.hpp
@@ -1287,10 +1287,22 @@
try
{
armnn::TensorInfo operandTensorInfo = GetTensorInfoForOperand(*operand);
+
if (IsDynamicTensor(operandTensorInfo))
{
- Fail("%s: dynamic input tensors are not supported", __func__);
- return LayerInputHandle();
+ const uint32_t operandIndex = operation.inputs[inputIndex];
+
+ // Check if the dynamic input tensors have been inferred by one of the previous layers
+ // If not we can't support them
+ if(data.m_OutputSlotForOperand.size() >= operandIndex && data.m_OutputSlotForOperand[operandIndex])
+ {
+ operandTensorInfo = data.m_OutputSlotForOperand[operandIndex]->GetTensorInfo();
+ }
+ else
+ {
+ Fail("%s: Type 2 dynamic input tensors are not supported", __func__);
+ return LayerInputHandle();
+ }
}
switch (operand->lifetime)