1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/Graph.cpp b/src/armnn/Graph.cpp
index 6943076ada..fee1da4343 100644
--- a/src/armnn/Graph.cpp
+++ b/src/armnn/Graph.cpp
@@ -52,6 +52,11 @@ Graph::Graph(const Graph& other)
                 Layer* const thisTgtLayer = otherToClonedMap[&otherTgtLayer];
 
                 InputSlot& inputSlot = thisTgtLayer->GetInputSlot(otherInputSlot->GetSlotIndex());
+
+                if (otherInputSlot->IsTensorInfoOverridden())
+                {
+                    inputSlot.SetTensorInfo(otherInputSlot->GetTensorInfo());
+                }
                 outputSlot->Connect(inputSlot);
             }
             outputSlot->SetTensorInfo(otherOutputSlot.GetTensorInfo());