From aa920c56838c2a0b31bd4e3c54bd57ff2f20969e Mon Sep 17 00:00:00 2001 From: Tee Jung 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 Change-Id: I25ceeca70e72fad88ab039aed5a5ab6a7cc08c6c Signed-off-by: Derek Lamberti --- src/armnnDeserializer/Deserializer.hpp | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'src/armnnDeserializer/Deserializer.hpp') diff --git a/src/armnnDeserializer/Deserializer.hpp b/src/armnnDeserializer/Deserializer.hpp index ae8be6e932..8e8fe1aca8 100644 --- a/src/armnnDeserializer/Deserializer.hpp +++ b/src/armnnDeserializer/Deserializer.hpp @@ -144,9 +144,21 @@ private: void SetupInputLayers(GraphPtr graphPtr); void SetupOutputLayers(GraphPtr graphPtr); + /// Helper to get the index of the layer in the flatbuffer vector from its bindingId property + unsigned int GetInputLayerInVector(GraphPtr graph, int targetId); + unsigned int GetOutputLayerInVector(GraphPtr graph, int targetId); + /// Helper to get the index of the layer in the flatbuffer vector from its index property unsigned int GetLayerIndexInVector(GraphPtr graph, unsigned int index); + struct FeatureVersions + { + // Default values to zero for backward compatibility + unsigned int m_BindingIdScheme = 0; + }; + + FeatureVersions GetFeatureVersions(GraphPtr graph); + /// The network we're building. Gets cleared after it is passed to the user armnn::INetworkPtr m_Network; std::vector m_ParserFunctions; -- cgit v1.2.1