1 files changed, 10 insertions, 0 deletions
diff --git a/src/armnnSerializer/test/SerializerTests.cpp b/src/armnnSerializer/test/SerializerTests.cpp
index c4c6eedb20..822f9c7e00 100644
--- a/src/armnnSerializer/test/SerializerTests.cpp
+++ b/src/armnnSerializer/test/SerializerTests.cpp
@@ -145,6 +145,12 @@ BOOST_AUTO_TEST_CASE(SerializeAddition)
     armnn::IConnectableLayer* const outputLayer0 = network->AddOutputLayer(0);
     additionLayer0->GetOutputSlot(0).Connect(outputLayer0->GetInputSlot(0));
 
+    armnn::TensorShape shape{1U};
+    armnn::TensorInfo info(shape, armnn::DataType::Float32);
+    inputLayer0->GetOutputSlot(0).SetTensorInfo(info);
+    inputLayer1->GetOutputSlot(0).SetTensorInfo(info);
+    additionLayer0->GetOutputSlot(0).SetTensorInfo(info);
+
     armnnSerializer::Serializer serializer;
     serializer.Serialize(*network);
 
@@ -170,6 +176,10 @@ BOOST_AUTO_TEST_CASE(SerializeMultiplication)
     armnn::IConnectableLayer* const outputLayer0 = network->AddOutputLayer(0);
     multiplicationLayer0->GetOutputSlot(0).Connect(outputLayer0->GetInputSlot(0));
 
+    inputLayer0->GetOutputSlot(0).SetTensorInfo(info);
+    inputLayer1->GetOutputSlot(0).SetTensorInfo(info);
+    multiplicationLayer0->GetOutputSlot(0).SetTensorInfo(info);
+
     armnnSerializer::Serializer serializer;
     serializer.Serialize(*network);