From 4dc64a69ba383ece509d442598617445a3b4847f Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Mon, 16 Sep 2019 17:00:22 +0100 Subject: 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 Change-Id: I7bc3d610dda9526190187eb87394a8ed7a4b5cdd --- src/backends/reference/workloads/CMakeLists.txt | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/backends/reference/workloads/CMakeLists.txt') diff --git a/src/backends/reference/workloads/CMakeLists.txt b/src/backends/reference/workloads/CMakeLists.txt index 7f49e800e5..23d6024530 100644 --- a/src/backends/reference/workloads/CMakeLists.txt +++ b/src/backends/reference/workloads/CMakeLists.txt @@ -8,6 +8,8 @@ list(APPEND armnnRefBackendWorkloads_sources Abs.hpp Activation.cpp Activation.hpp + ArgMinMax.cpp + ArgMinMax.hpp BaseIterator.hpp BatchNormImpl.cpp BatchNormImpl.hpp -- cgit v1.2.1