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 /src/armnn/Layer.cpp | |
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
Diffstat (limited to 'src/armnn/Layer.cpp')
-rw-r--r-- | src/armnn/Layer.cpp | 7 |
1 files changed, 7 insertions, 0 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 |