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>
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index d92c93d..14d2776 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -29,12 +29,12 @@
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 @@
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 @@
/// 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;