aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/Types.hpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-09-09 15:01:18 +0100
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-09-09 15:10:50 +0100
commit8d001d40f4d45e9df16f913885d7f6dbaa179c0f (patch)
tree0650f0fce6c654bca54411c32eb88c82f79ba92c /include/armnn/Types.hpp
parentbd707161fe5d5c92540b693bcbb8256fe620b5c2 (diff)
downloadarmnn-8d001d40f4d45e9df16f913885d7f6dbaa179c0f.tar.gz
IVGCVSW-3722 Add function to specify min or max in ArgMinMaxDescriptor
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com> Change-Id: I2f9d9968cbfda3ec3eadbe9a746e93e816989a3d
Diffstat (limited to 'include/armnn/Types.hpp')
-rw-r--r--include/armnn/Types.hpp6
1 files changed, 6 insertions, 0 deletions
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp
index 6d8ea6f877..dbcb91aebe 100644
--- a/include/armnn/Types.hpp
+++ b/include/armnn/Types.hpp
@@ -54,6 +54,12 @@ enum class ActivationFunction
Square = 9
};
+enum class ArgMinMaxFunction
+{
+ Min = 0,
+ Max = 1
+};
+
enum class PoolingAlgorithm
{
Max = 0,