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