From aa920c56838c2a0b31bd4e3c54bd57ff2f20969e Mon Sep 17 00:00:00 2001
From: Tee Jung <tee.ty.jung@openedges.com>
Date: Tue, 5 Nov 2019 10:48:25 +0000
Subject: Build graph->inputIds/outputIds with layerBindingId instead of
 layerIndex

Signed-off-by: Jung Tae-young tee.ty.jung@openedges.com
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Change-Id: I25ceeca70e72fad88ab039aed5a5ab6a7cc08c6c
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
---
 src/armnnSerializer/Serializer.hpp | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

(limited to 'src/armnnSerializer/Serializer.hpp')

diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index d92c93d46c..14d2776147 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -29,12 +29,12 @@ public:
         return m_flatBufferBuilder;
     }
 
-    std::vector<uint32_t>& GetInputIds()
+    std::vector<int>& GetInputIds()
     {
         return m_inputIds;
     }
 
-    std::vector<uint32_t>& GetOutputIds()
+    std::vector<int>& GetOutputIds()
     {
         return m_outputIds;
     }
@@ -44,6 +44,9 @@ public:
         return m_serializedLayers;
     }
 
+    flatbuffers::Offset<armnnSerializer::FeatureCompatibilityVersions> GetVersionTable();
+
+
     ARMNN_DEPRECATED_MSG("Use VisitElementwiseUnaryLayer instead")
     void VisitAbsLayer(const armnn::IConnectableLayer* layer,
                        const char* name = nullptr) override;
@@ -301,11 +304,11 @@ private:
     /// AnyLayers required by the SerializedGraph.
     std::vector<flatbuffers::Offset<armnnSerializer::AnyLayer>> m_serializedLayers;
 
-    /// Vector of indexes of all Input Layers required by the SerializedGraph.
-    std::vector<uint32_t> m_inputIds;
+    /// Vector of the binding ids of all Input Layers required by the SerializedGraph.
+    std::vector<int> m_inputIds;
 
-    /// Vector of indexes of all Output Layers required by the SerializedGraph.
-    std::vector<uint32_t> m_outputIds;
+    /// Vector of the binding ids of all Output Layers required by the SerializedGraph.
+    std::vector<int> m_outputIds;
 
     /// Mapped Guids of all Layers to match our index.
     std::unordered_map<armnn::LayerGuid, uint32_t > m_guidMap;
-- 
cgit v1.2.1