aboutsummaryrefslogtreecommitdiff
path: root/src/armnnDeserializer/Deserializer.hpp
diff options
context:
space:
mode:
authorTee Jung <tee.ty.jung@openedges.com>2019-11-05 10:48:25 +0000
committerDerek Lamberti <derek.lamberti@arm.com>2020-02-04 13:50:36 +0000
commitaa920c56838c2a0b31bd4e3c54bd57ff2f20969e (patch)
treed029c5685fd3a622fd94c2e9707821aa0b47ca06 /src/armnnDeserializer/Deserializer.hpp
parent9a61fa6bdc8724cc66eba532360e0e48b6b1a60e (diff)
downloadarmnn-experimental/QuantizerStuff.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.hpp12
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;