diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 9630d86197..87f4bdb40e 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -30,6 +30,15 @@ struct ActivationDescriptor float m_B; }; +/// An ArgMinMaxDescriptor for ArgMinMaxLayer +struct ArgMinMaxDescriptor +{ + ArgMinMaxDescriptor() + : m_Axis(-1) {} + + int m_Axis; +}; + /// A PermuteDescriptor for the PermuteLayer. struct PermuteDescriptor { |