From 3091b06807257a6db0ff03654681f87526f37955 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 15 Feb 2019 14:45:04 +0000 Subject: 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 --- src/armnnSerializer/test/SerializerTests.cpp | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'src/armnnSerializer') 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); -- cgit v1.2.1