diff options
| author | Mike Kelly <mike.kelly@arm.com> | 2020-11-17 13:55:01 +0000 |
|---|---|---|
| committer | Jim Flynn <jim.flynn@arm.com> | 2020-11-17 15:38:07 +0000 |
| commit | 7a0efa51aca2bc0cf3aaefe22e90525ef7cc2478 (patch) | |
| tree | 500323a71b611099bbf6b857e3eb942b7b8936c3 | |
| parent | a8578103d1fe621f97ff2cfd842a8e33c1b652c8 (diff) | |
| download | armnn-7a0efa51aca2bc0cf3aaefe22e90525ef7cc2478.tar.gz | |
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
| -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); } |