1 files changed, 5 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index cf098eb5f8..e10b66f51d 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -1974,12 +1974,16 @@ std::vector<fb::Offset<serializer::InputSlot>>
 
         // Get the Connection for the InputSlot
         const IOutputSlot* connection = inputSlot.GetConnection();
+        bool isOverridden = inputSlot.IsTensorInfoOverridden();
+
+        flatbuffers::Offset<TensorInfo> overriddenTensorInfo = CreateTensorInfo(inputSlot.GetTensorInfo());
 
         // Create FlatBuffer Connection
         serializer::Connection conn(GetSerializedId(inputSlot.GetConnection()->GetOwningLayerGuid()),
                                     connection->CalculateIndexOnOwner());
         // Create FlatBuffer InputSlot
-        inputSlots.push_back(serializer::CreateInputSlot(m_flatBufferBuilder, slotIndex, &conn));
+        inputSlots.push_back(serializer::CreateInputSlot(m_flatBufferBuilder, slotIndex, &conn, isOverridden,
+                                                         overriddenTensorInfo));
     }
     return inputSlots;
 }