aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/Descriptors.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r--include/armnn/Descriptors.hpp24
1 files changed, 24 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index 9ff894f1b0..27ca50123f 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -1620,4 +1620,28 @@ struct BatchMatMulDescriptor : BaseDescriptor
const TensorShape& tensorShape);
};
+struct ReverseV2Descriptor : BaseDescriptor
+{
+ ReverseV2Descriptor()
+ : m_Axis()
+ , m_MaxDimension(4)
+ {}
+
+ ReverseV2Descriptor(std::vector<int32_t> axis)
+ : m_Axis(axis)
+ , m_MaxDimension(4)
+ {}
+
+ bool operator ==(const ReverseV2Descriptor& rhs) const
+ {
+ return m_Axis == rhs.m_Axis;
+ }
+
+ /// The indices of the dimensions to reverse
+ std::vector<int32_t> m_Axis;
+ /// The max dimension supported in the lower levels of code
+ uint32_t m_MaxDimension;
+
+};
+
} // namespace armnn