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 9ff894f1b0..3a571a66e8 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -1620,4 +1620,23 @@ struct BatchMatMulDescriptor : BaseDescriptor const TensorShape& tensorShape); }; +struct TileDescriptor : BaseDescriptor +{ + TileDescriptor() + : m_Multiples() + {} + + explicit TileDescriptor(std::vector<uint32_t> multiples) + : m_Multiples(std::move(multiples)) + {} + + bool operator ==(const TileDescriptor& rhs) const + { + return m_Multiples == rhs.m_Multiples; + } + + /// The vector to multiply the input shape by + std::vector<uint32_t> m_Multiples; +}; + } // namespace armnn |