diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 30eaefd83b..bf40b35ae9 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -1656,4 +1656,23 @@ struct TileDescriptor : BaseDescriptor std::vector<uint32_t> m_Multiples; }; +struct BroadcastToDescriptor : BaseDescriptor +{ + BroadcastToDescriptor() + : m_BroadcastToShape() + {} + + explicit BroadcastToDescriptor(const TensorShape& shape) + : m_BroadcastToShape(shape) + {} + + bool operator ==(const BroadcastToDescriptor& rhs) const + { + return m_BroadcastToShape == rhs.m_BroadcastToShape; + } + + /// Target shape value. + TensorShape m_BroadcastToShape; +}; + } // namespace armnn |