projects / platform / upstream / armnn.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: a0c7871) | patch
IVGCVSW-3694 Add ArgMinMax implementation for Ref
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>
Mon, 16 Sep 2019 16:00:22 +0000 (17:00 +0100)
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>
Mon, 16 Sep 2019 16:00:54 +0000 (17:00 +0100)
commit4dc64a69ba383ece509d442598617445a3b4847f
treeb50cb259594aa0cf634a4c37657a2c7a50be0c6ctree | snapshot
parenta0c7871cf140d1e9cf59a213626ee534c0122c7fcommit | diff
IVGCVSW-3694 Add ArgMinMax implementation for Ref

 * Add ArgMinMax implementation
 * Add utility function to get number of elements between axis
 * Add utility function to get unsigned axis
 * Unit tests for ArgMinMax function

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I7bc3d610dda9526190187eb87394a8ed7a4b5cdd
src/armnnUtils/TensorUtils.cpp diff | blob | history
src/armnnUtils/TensorUtils.hpp diff | blob | history
src/backends/reference/test/ArgMinMaxTests.cpp [new file with mode: 0644] blob
src/backends/reference/test/CMakeLists.txt diff | blob | history
src/backends/reference/workloads/ArgMinMax.cpp [new file with mode: 0644] blob
src/backends/reference/workloads/ArgMinMax.hpp [new file with mode: 0644] blob
src/backends/reference/workloads/CMakeLists.txt diff | blob | history
src/backends/reference/workloads/Softmax.cpp diff | blob | history
Domain: Machine Learning / ML Framework;