diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
-rw-r--r-- | include/armnn/Descriptors.hpp | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index decbf99880..5f9df6b2c3 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -332,4 +332,19 @@ struct LstmDescriptor bool m_ProjectionEnabled; }; +struct MeanDescriptor +{ + MeanDescriptor() + : m_KeepDims(false) + {} + + MeanDescriptor(const std::vector<unsigned int>& axis, bool keepDims) + : m_Axis(axis) + , m_KeepDims(keepDims) + {} + + std::vector<unsigned int> m_Axis; + bool m_KeepDims; +}; + } |