IVGCVSW-2610 Add Quantization Tool Executable

* Also added TensorInfos to the output slots in the SerializerTests
  to create valid ArmNN networks for serialization

Change-Id: I092b1ac889dd5e05cb1c10c9dfb573acaf1970d9
Signed-off-by: Jim Flynn <jim.flynn@arm.com>

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);