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 ba9a56ad38..45c0f421f3 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -78,6 +78,26 @@ struct ComparisonDescriptor ComparisonOperation m_Operation; }; +/// A ElementwiseUnaryDescriptor for the ElementwiseUnaryLayer +struct ElementwiseUnaryDescriptor +{ + ElementwiseUnaryDescriptor() + : ElementwiseUnaryDescriptor(UnaryOperation::Abs) + {} + + ElementwiseUnaryDescriptor(UnaryOperation operation) + : m_Operation(operation) + {} + + bool operator ==(const ElementwiseUnaryDescriptor &rhs) const + { + return m_Operation == rhs.m_Operation; + } + + /// Specifies the elementwiseUnary operation to execute + UnaryOperation m_Operation; +}; + /// A PermuteDescriptor for the PermuteLayer. struct PermuteDescriptor { |