diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-02-15 14:45:04 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-02-21 08:26:39 +0000 |
commit | 3091b06807257a6db0ff03654681f87526f37955 (patch) | |
tree | 30855ced21aa74400d6910040e300d3eda15f269 /src/armnnSerializer | |
parent | 30b0020478652e441a5dff4880261f7c7007bb6c (diff) | |
download | armnn-3091b06807257a6db0ff03654681f87526f37955.tar.gz |
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>
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r-- | src/armnnSerializer/test/SerializerTests.cpp | 10 |
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); |