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.hpp15
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;
+};
+
}