aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/Descriptors.hpp
diff options
context:
space:
mode:
authornarpra01 <narumol.prangnawarat@arm.com>2018-09-13 11:07:48 +0100
committerMatthew Bentham <matthew.bentham@arm.com>2018-10-01 14:56:48 +0100
commit32b9046ea74d2387a08819cf5e67c183e03f6d3f (patch)
tree87d1627aed07bc1c572d7b2121974e24778fe81a /include/armnn/Descriptors.hpp
parent5cd01f35e8975ca6d42eb5f2d5f121b57bdb8086 (diff)
downloadarmnn-32b9046ea74d2387a08819cf5e67c183e03f6d3f.tar.gz
IVGCVSW-1813 - Add MeanLayer
* add MeanLayer functionalities * modify MeanQueueDescriptor to use parameter * add IsMeanSupported placeholder for all backends Change-Id: Ic69a34a61df667849977aad9b38f9a01eef565b5
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;
+};
+
}