From 85d3671618f0d40b71ebbc80373389140390c2cd Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Tue, 26 Jan 2021 22:30:06 +0000 Subject: IVGCVSW-5593 Implement Pimpl Idiom for serialization classes Signed-off-by: Finn Williams Change-Id: I01c7bd314219e58f71505dcb787d606dbded914a --- src/armnnSerializer/test/ActivationSerializationTests.cpp | 7 ++++--- src/armnnSerializer/test/SerializerTests.cpp | 7 ++++--- 2 files changed, 8 insertions(+), 6 deletions(-) (limited to 'src/armnnSerializer/test') diff --git a/src/armnnSerializer/test/ActivationSerializationTests.cpp b/src/armnnSerializer/test/ActivationSerializationTests.cpp index abc63ae64d..1645731413 100644 --- a/src/armnnSerializer/test/ActivationSerializationTests.cpp +++ b/src/armnnSerializer/test/ActivationSerializationTests.cpp @@ -54,11 +54,12 @@ BOOST_AUTO_TEST_CASE(ActivationSerialization) activationLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0)); activationLayer->GetOutputSlot(0).SetTensorInfo(outputInfo); - armnnSerializer::Serializer serializer; - serializer.Serialize(*network); + armnnSerializer::ISerializerPtr serializer = armnnSerializer::ISerializer::Create(); + + serializer->Serialize(*network); std::stringstream stream; - serializer.SaveSerializedToStream(stream); + serializer->SaveSerializedToStream(stream); std::string const serializerString{stream.str()}; std::vector const serializerVector{serializerString.begin(), serializerString.end()}; diff --git a/src/armnnSerializer/test/SerializerTests.cpp b/src/armnnSerializer/test/SerializerTests.cpp index 6866391e0f..11177f5d04 100644 --- a/src/armnnSerializer/test/SerializerTests.cpp +++ b/src/armnnSerializer/test/SerializerTests.cpp @@ -237,11 +237,12 @@ armnn::INetworkPtr DeserializeNetwork(const std::string& serializerString) std::string SerializeNetwork(const armnn::INetwork& network) { - armnnSerializer::Serializer serializer; - serializer.Serialize(network); + armnnSerializer::ISerializerPtr serializer = armnnSerializer::ISerializer::Create(); + + serializer->Serialize(network); std::stringstream stream; - serializer.SaveSerializedToStream(stream); + serializer->SaveSerializedToStream(stream); std::string serializerString{stream.str()}; return serializerString; -- cgit v1.2.1