From 07f2121feeeeae36a7e67eeb8a6965df63b848f3 Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Tue, 23 Jul 2019 09:50:50 +0100 Subject: IVGCVSW-3536 Add Axis parameter to reference Softmax implementation * Add Axis parameter to Softmax Descriptor * Add new reference implementation for Softmax using Axis parameter * Add unit tests to cover each Axis Change-Id: Iafac2275d2212337456f2b1b56b0f76f77fb9543 Signed-off-by: Francis Murtagh --- include/armnn/Descriptors.hpp | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'include/armnn/Descriptors.hpp') diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 377f0705d7..9630d86197 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -49,9 +49,15 @@ struct PermuteDescriptor /// A SoftmaxDescriptor for the SoftmaxLayer. struct SoftmaxDescriptor { - SoftmaxDescriptor() : m_Beta(1.0f) {} + SoftmaxDescriptor() + : m_Beta(1.0f) + , m_Axis(-1) + {} + /// Exponentiation value. - float m_Beta; + float m_Beta; + /// Scalar, defaulted to the last index (-1), specifying the dimension the activation will be performed on. + int m_Axis; }; /// @brief An OriginsDescriptor for the ConcatLayer. -- cgit v1.2.1