diff options
author | Tee Jung <tee.ty.jung@openedges.com> | 2019-11-05 10:48:25 +0000 |
---|---|---|
committer | Derek Lamberti <derek.lamberti@arm.com> | 2020-02-04 13:50:36 +0000 |
commit | aa920c56838c2a0b31bd4e3c54bd57ff2f20969e (patch) | |
tree | d029c5685fd3a622fd94c2e9707821aa0b47ca06 /src/armnnDeserializer/Deserializer.hpp | |
parent | 9a61fa6bdc8724cc66eba532360e0e48b6b1a60e (diff) | |
download | armnn-aa920c56838c2a0b31bd4e3c54bd57ff2f20969e.tar.gz |
Build graph->inputIds/outputIds with layerBindingId instead of layerIndexexperimental/QuantizerStuff
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>
Diffstat (limited to 'src/armnnDeserializer/Deserializer.hpp')
-rw-r--r-- | src/armnnDeserializer/Deserializer.hpp | 12 |
1 files changed, 12 insertions, 0 deletions
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<LayerParsingFunction> m_ParserFunctions; |