diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 92e842b2c1..10d8ab7a08 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -58,6 +58,26 @@ struct ArgMinMaxDescriptor int m_Axis; }; +/// A ComparisonDescriptor for the ComparisonLayer +struct ComparisonDescriptor +{ + ComparisonDescriptor() + : ComparisonDescriptor(ComparisonOperation::Equal) + {} + + ComparisonDescriptor(ComparisonOperation operation) + : m_Operation(operation) + {} + + bool operator ==(const ComparisonDescriptor &rhs) const + { + return m_Operation == rhs.m_Operation; + } + + /// Specifies the comparison operation to execute + ComparisonOperation m_Operation; +}; + /// A PermuteDescriptor for the PermuteLayer. struct PermuteDescriptor { |