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/armnnSerializer/Serializer.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/armnnSerializer/Serializer.hpp')
-rw-r--r-- | src/armnnSerializer/Serializer.hpp | 15 |
1 files changed, 9 insertions, 6 deletions
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; |