diff options
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, |