MLCE-1093 Added Axis to ViewsDescriptor

 * Added Axis to ViewsDescriptor to store the value where ever possible.
 * Updated Serializer and Deserializer to handle axis.

Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I56e442872b47485a608b25fbc79063b362a25618

1 files changed, 12 insertions, 0 deletions

diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index f1ac17f4c6..f60e8f3bea 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -276,9 +276,21 @@ struct ViewsDescriptor : BaseDescriptor
 
     /// Swap the ViewsDescriptor value first and second.
     friend void swap(ViewsDescriptor& first, ViewsDescriptor& second);
+
+    /// Set the axis value.
+    void SetAxis(int32_t axis);
+
+    /// Get the axis value.
+    int32_t GetAxis() const;
+
+    /// Returns true if an axis has been set.
+    bool HasAxis() const;
+
 private:
     OriginsDescriptor m_Origins;
     uint32_t**        m_ViewSizes;
+    bool              m_IsAxisSet = false;
+    int32_t           m_Axis = 0;
 };