diff options
-rw-r--r-- | src/armnn/Layer.cpp | 7 | ||||
-rw-r--r-- | src/armnn/Layer.hpp | 2 |
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); } |