diff options
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 |