aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/armnn/Layer.cpp7
-rw-r--r--src/armnn/Layer.hpp2
2 files changed, 8 insertions, 1 deletions
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);
}