diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2021-11-08 17:18:31 +0000 |
|---|---|---|
| committer | David Monahan <david.monahan@arm.com> | 2021-11-08 20:05:43 +0000 |
| commit | fd0cae3153dd7844b1b14adda8571e4588ed762c (patch) | |
| tree | 03995f2f6395c4a8f8152ede6eccc5ab3bad4b59 /src/armnnSerializer/Serializer.cpp | |
| parent | 5b8093c17044e8eaaaa42d96ba4902dee5791be4 (diff) | |
| download | armnn-fd0cae3153dd7844b1b14adda8571e4588ed762c.tar.gz | |
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
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
| -rw-r--r-- | src/armnnSerializer/Serializer.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index c08784352d..a9f7b7d307 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -1037,7 +1037,9 @@ void SerializerStrategy::SerializeSoftmaxLayer(const armnn::IConnectableLayer* l // 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 = |