1 files changed, 7 insertions, 0 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp
index e940bd69ec..d125ef56dd 100644
--- a/src/armnnDeserializer/Deserializer.cpp
+++ b/src/armnnDeserializer/Deserializer.cpp
@@ -1156,6 +1156,13 @@ void IDeserializer::DeserializerImpl::RegisterInputSlots(GraphPtr graph,
             auto fbInputSlot = baseLayer->inputSlots()->Get(i);
             auto fbConnection = fbInputSlot->connection();
             armnn::IInputSlot* inputSlot = &(layer->GetInputSlot(fbInputSlot->index()));
+
+            // If the slot has an Overridden tensorInfo then extract it
+            if (fbInputSlot->isOverridden())
+            {
+                armnn::TensorInfo overriddenTensorInfo = ToTensorInfo(fbInputSlot->overriddenTensorInfo());
+                inputSlot->SetTensorInfo(overriddenTensorInfo);
+            }
             RegisterInputSlotOfConnection(fbConnection->sourceLayerIndex(), fbConnection->outputSlotIndex(), inputSlot);
         }
     }