commit fd0cae3153dd7844b1b14adda8571e4588ed762c
author Sadik Armagan <sadik.armagan@arm.com> Mon Nov 08 17:18:31 2021 +0000
committer David Monahan <david.monahan@arm.com> Mon Nov 08 20:05:43 2021 +0000
tree 03995f2f6395c4a8f8152ede6eccc5ab3bad4b59
parent 5b8093c17044e8eaaaa42d96ba4902dee5791be4

IVGCVSW-5636 'Implement NNAPI caching functions'

* Serialize/Deserialize axis parameter on Softmax Descriptor.

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I1bba0dd7d78f9f211f77115dcefccf24a4a61b39

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index c087843..a9f7b7d 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -1037,7 +1037,9 @@
 
     // Create the FlatBuffer SoftmaxDescriptor
     auto flatBufferSoftmaxDesc =
-        serializer::CreateSoftmaxDescriptor(m_flatBufferBuilder, softmaxDescriptor.m_Beta);
+        serializer::CreateSoftmaxDescriptor(m_flatBufferBuilder,
+                                            softmaxDescriptor.m_Beta,
+                                            softmaxDescriptor.m_Axis);
 
     // Create the FlatBuffer SoftmaxLayer
     auto flatBufferSoftmaxLayer =