projects / platform / upstream / armnn.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 82b15ed) | patch
IVGCVSW-3554 Update workloads to pass Softmax Axis Parameter to Backends
authorFrancis Murtagh <francis.murtagh@arm.com>
Fri, 26 Jul 2019 14:44:17 +0000 (15:44 +0100)
committerFrancis Murtagh <francis.murtagh@arm.com>
Fri, 26 Jul 2019 14:44:17 +0000 (15:44 +0100)
commit3b9383548309a8f9121668826a628ec250c57a1c
tree01cc4e162008aea23bfe9ae167ded0e8f9abc39ctree | snapshot
parent82b15eda6f87a20bc31256f5e85eb4082d3d0591commit | diff
IVGCVSW-3554 Update workloads to pass Softmax Axis Parameter to Backends

 * Add check in CL and Neon to ensure axis is 1 otherwise
   return unsupported.
 * Edit CreateWorkload test and JsonPrinter test to ensure axis of 1.

Change-Id: I499b405532e26fefc2dd1c18b6dc6005813b5604
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
src/armnn/test/CreateWorkload.hpp diff | blob | history
src/armnnCaffeParser/CaffeParser.cpp diff | blob | history
src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp diff | blob | history
src/backends/cl/ClLayerSupport.cpp diff | blob | history
src/backends/cl/workloads/ClSoftmaxBaseWorkload.cpp diff | blob | history
src/backends/cl/workloads/ClSoftmaxBaseWorkload.hpp diff | blob | history
src/backends/cl/workloads/ClSoftmaxFloatWorkload.cpp diff | blob | history
src/backends/neon/NeonLayerSupport.cpp diff | blob | history
src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp diff | blob | history
src/backends/neon/workloads/NeonSoftmaxFloatWorkload.cpp diff | blob | history
src/backends/neon/workloads/NeonSoftmaxUint8Workload.cpp diff | blob | history
Domain: Machine Learning / ML Framework;