From 7a0efa51aca2bc0cf3aaefe22e90525ef7cc2478 Mon Sep 17 00:00:00 2001
From: Mike Kelly <mike.kelly@arm.com>
Date: Tue, 17 Nov 2020 13:55:01 +0000
Subject: IVGCVSW-5535 Extend dump file with info about fused layers

 * Add optional ActivationDescriptor information to SerializeLayerParameters

Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I6268932cdc4637cdb30948e1b7f0f0649ba18492
---
 src/armnn/Layer.cpp | 7 +++++++
 src/armnn/Layer.hpp | 2 +-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/armnn/Layer.cpp b/src/armnn/Layer.cpp
index 62f861b4a0..9a526a0943 100644
--- a/src/armnn/Layer.cpp
+++ b/src/armnn/Layer.cpp
@@ -464,6 +464,13 @@ void Layer::SerializeLayerParameters(ParameterStringifyFunction& fn) const
     {
         fn("BackendID",backendId);
     }
+    std::shared_ptr<ActivationDescriptor>
+            activationDescPtr = GetAdditionalInformation<ActivationDescriptor>();
+
+    if (activationDescPtr)
+    {
+        StringifyLayerParameters<ActivationDescriptor>::Serialize(fn, *activationDescPtr.get());
+    }
 }
 
 } // namespace armnn
diff --git a/src/armnn/Layer.hpp b/src/armnn/Layer.hpp
index 6cb3b8f5bd..ef0f8c3297 100644
--- a/src/armnn/Layer.hpp
+++ b/src/armnn/Layer.hpp
@@ -336,7 +336,7 @@ public:
     }
 
     template<typename T>
-    std::shared_ptr<T> GetAdditionalInformation()
+    std::shared_ptr<T> GetAdditionalInformation() const
     {
         return std::static_pointer_cast<T>(m_AdditionalInfoObject);
     }
-- 
cgit v1.2.1