projects / platform / upstream / armnn.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 6f3785d) | patch
IVGCVSW-3536 Add Axis parameter to reference Softmax implementation
authorFrancis Murtagh <francis.murtagh@arm.com>
Tue, 23 Jul 2019 08:50:50 +0000 (09:50 +0100)
committerFrancis Murtagh <francis.murtagh@arm.com>
Tue, 23 Jul 2019 08:50:56 +0000 (09:50 +0100)
commit07f2121feeeeae36a7e67eeb8a6965df63b848f3
tree836ebdec3ca1305f6533bde3f6410780a5173781tree | snapshot
parent6f3785d4f612e06854ab63dffbd2cd3d484c2e14commit | diff
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 <francis.murtagh@arm.com>
include/armnn/Descriptors.hpp diff | blob | history
src/backends/backendsCommon/test/LayerTests.cpp diff | blob | history
src/backends/backendsCommon/test/LayerTests.hpp diff | blob | history
src/backends/backendsCommon/test/SoftmaxTestImpl.hpp diff | blob | history
src/backends/reference/test/RefLayerTests.cpp diff | blob | history
src/backends/reference/workloads/RefSoftmaxWorkload.cpp diff | blob | history
src/backends/reference/workloads/Softmax.cpp diff | blob | history
src/backends/reference/workloads/Softmax.hpp diff | blob | history
Domain: Machine Learning / ML Framework;