aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/Descriptors.hpp
diff options
context:
space:
mode:
authorSimon Obute <simon.obute@arm.com>2021-09-03 15:50:13 +0100
committerTeresa Charlin <teresa.charlinreyes@arm.com>2021-09-24 16:06:30 +0100
commit51f67776a695c217a32596af806afeeb080f5528 (patch)
tree33ccfd87ba365bcc6fc86d5a2181991a130b3061 /include/armnn/Descriptors.hpp
parentf10b15a8946f39bdf3f60cebc59d2963069eedca (diff)
downloadarmnn-51f67776a695c217a32596af806afeeb080f5528.tar.gz
IVGCVSW-3705 Add Channel Shuffle Front end and Ref Implementation
* Add front end * Add reference workload * Add unit tests * Add Serializer and Deserializer * Update ArmNN Versioning Signed-off-by: Simon Obute <simon.obute@arm.com> Change-Id: I9ac1f953af3974382eac8e8d62d794d2344e8f47
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r--include/armnn/Descriptors.hpp22
1 files changed, 22 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index 341dbecd4f..d571f2297b 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -1342,4 +1342,26 @@ struct ReduceDescriptor : BaseDescriptor
ReduceOperation m_ReduceOperation;
};
+/// A ChannelShuffleDescriptor for the ChannelShuffle operator
+struct ChannelShuffleDescriptor : BaseDescriptor
+{
+ ChannelShuffleDescriptor()
+ : m_NumGroups(0), m_Axis(0)
+ {}
+
+ ChannelShuffleDescriptor(const uint32_t& numGroups, const uint32_t& axis)
+ : m_NumGroups(numGroups), m_Axis(axis)
+ {}
+
+ bool operator ==(const ChannelShuffleDescriptor& rhs) const
+ {
+ return m_NumGroups == rhs.m_NumGroups;
+ }
+
+ /// Number of groups for the channel shuffle operation
+ uint32_t m_NumGroups;
+ /// Axis to apply channel shuffle operation on
+ uint32_t m_Axis;
+};
+
} // namespace armnn