diff options
author | Saoirse Stewart <saoirse.stewart@arm.com> | 2019-02-14 15:46:10 +0000 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2019-02-15 13:12:05 +0000 |
commit | cb8a3219331c8eb89dcdfbcdb2569c0e9ecdcf4e (patch) | |
tree | 92fbf4bdecad2ee72ab4ca900a60d55e5f838cc9 /src/armnnSerializer/Serializer.hpp | |
parent | 3fcef20ff9b6fec5b853bc8166db8b9ae88f9f39 (diff) | |
download | armnn-cb8a3219331c8eb89dcdfbcdb2569c0e9ecdcf4e.tar.gz |
IVGCVSW-2642 Fix Guid issue in Serialize Parser
Change-Id: Id8ba083b6a5dee64ed197d71647ad3b451e4f354
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r-- | src/armnnSerializer/Serializer.hpp | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index b799dbe380..8a509e880e 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -10,6 +10,8 @@ #include <armnnSerializer/ISerializer.hpp> #include <iostream> +#include <unordered_map> + #include <Schema_generated.h> namespace armnnSerializer @@ -18,7 +20,7 @@ namespace armnnSerializer class SerializerVisitor : public armnn::LayerVisitorBase<armnn::VisitorNoThrowPolicy> { public: - SerializerVisitor() {} + SerializerVisitor() : m_layerId(0) {}; ~SerializerVisitor() {} flatbuffers::FlatBufferBuilder& GetFlatBufferBuilder() @@ -65,6 +67,9 @@ private: /// Creates the serializer AnyLayer for the layer and adds it to m_serializedLayers. void CreateAnyLayer(const flatbuffers::Offset<void>& layer, const armnn::armnnSerializer::Layer serializerLayer); + ///Function which maps Guid to an index + uint32_t GetSerializedId(unsigned int guid); + /// Creates the serializer InputSlots for the layer. std::vector<flatbuffers::Offset<armnn::armnnSerializer::InputSlot>> CreateInputSlots( const armnn::IConnectableLayer* layer); @@ -84,6 +89,12 @@ private: /// Guids of all Output Layers required by the SerializedGraph. std::vector<unsigned int> m_outputIds; + + /// Mapped Guids of all Layers to match our index. + std::unordered_map<unsigned int, uint32_t > m_guidMap; + + /// layer within our FlatBuffer index. + uint32_t m_layerId; }; class Serializer : public ISerializer |