diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-09-09 15:01:18 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-09-09 15:10:50 +0100 |
| commit | 8d001d40f4d45e9df16f913885d7f6dbaa179c0f (patch) | |
| tree | 0650f0fce6c654bca54411c32eb88c82f79ba92c /include | |
| parent | bd707161fe5d5c92540b693bcbb8256fe620b5c2 (diff) | |
| download | armnn-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')
| -rw-r--r-- | include/armnn/Descriptors.hpp | 6 | ||||
| -rw-r--r-- | include/armnn/Types.hpp | 6 |
2 files changed, 11 insertions, 1 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 87f4bdb40e..e871e89ebb 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -34,8 +34,12 @@ struct ActivationDescriptor struct ArgMinMaxDescriptor { ArgMinMaxDescriptor() - : m_Axis(-1) {} + : m_Function(ArgMinMaxFunction::Min) + , m_Axis(-1) {} + /// Specify if the function is to find Min or Max. + ArgMinMaxFunction m_Function; + /// Axis to reduce across the input tensor. int m_Axis; }; 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, |