diff options
author | Éanna Ó Catháin <eanna.ocathain@arm.com> | 2019-02-25 16:26:29 +0000 |
---|---|---|
committer | Eanna O Cathain Arm <eanna.ocathain@arm.com> | 2019-02-25 17:28:08 +0000 |
commit | 633f859043522adb7629f3f7ea617cd65b3713f2 (patch) | |
tree | 39fdc2fdc5709076c1852f85ee8aa6702c83dae6 /src/armnnSerializer/test/ActivationSerializationTests.cpp | |
parent | 268509ac1750c6c8d7c6f0debd9dca7e42612341 (diff) | |
download | armnn-633f859043522adb7629f3f7ea617cd65b3713f2.tar.gz |
IVGCVSW-2751 When layers are Deserialized the layer name is lost
* Fixed defect
* Added unit tests to check names
Change-Id: I67211110b90aabff6e00bccdadd1145423b07523
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Signed-off-by: Éanna Ó Catháin <eanna.ocathain@arm.com>
Diffstat (limited to 'src/armnnSerializer/test/ActivationSerializationTests.cpp')
-rw-r--r-- | src/armnnSerializer/test/ActivationSerializationTests.cpp | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/armnnSerializer/test/ActivationSerializationTests.cpp b/src/armnnSerializer/test/ActivationSerializationTests.cpp index c20f2864f9..2a46045f59 100644 --- a/src/armnnSerializer/test/ActivationSerializationTests.cpp +++ b/src/armnnSerializer/test/ActivationSerializationTests.cpp @@ -12,6 +12,17 @@ BOOST_AUTO_TEST_SUITE(SerializerTests) +class VerifyActivationName : public armnn::LayerVisitorBase<armnn::VisitorNoThrowPolicy> +{ +public: + void VisitActivationLayer(const armnn::IConnectableLayer* layer, + const armnn::ActivationDescriptor& activationDescriptor, + const char* name) override + { + BOOST_TEST(name == "activation"); + } +}; + BOOST_AUTO_TEST_CASE(ActivationSerialization) { armnnDeserializer::IDeserializerPtr parser = armnnDeserializer::IDeserializer::Create(); @@ -48,6 +59,9 @@ BOOST_AUTO_TEST_CASE(ActivationSerialization) armnn::INetworkPtr deserializedNetwork = parser->CreateNetworkFromBinary(serializerVector); + VerifyActivationName visitor; + deserializedNetwork->Accept(visitor); + armnn::IRuntime::CreationOptions options; // default options armnn::IRuntimePtr run = armnn::IRuntime::Create(options); auto deserializedOptimized = Optimize(*deserializedNetwork, { armnn::Compute::CpuRef }, run->GetDeviceSpec()); |