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