diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 7c6942ce58..9ff894f1b0 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -105,6 +105,26 @@ struct ComparisonDescriptor : BaseDescriptor ComparisonOperation m_Operation; }; +/// A ElementwiseBinaryDescriptor for the ElementwiseBinaryLayer +struct ElementwiseBinaryDescriptor : BaseDescriptor +{ + ElementwiseBinaryDescriptor() + : ElementwiseBinaryDescriptor(BinaryOperation::Add) + {} + + ElementwiseBinaryDescriptor(BinaryOperation operation) + : m_Operation(operation) + {} + + bool operator ==(const ElementwiseBinaryDescriptor &rhs) const + { + return m_Operation == rhs.m_Operation; + } + + /// Specifies the elementwiseBinary operation to execute + BinaryOperation m_Operation; +}; + /// A ElementwiseUnaryDescriptor for the ElementwiseUnaryLayer struct ElementwiseUnaryDescriptor : BaseDescriptor { |