diff options
author | janeil01 <jan.eilers@arm.com> | 2019-11-07 09:47:20 +0000 |
---|---|---|
committer | janeil01 <jan.eilers@arm.com> | 2019-11-08 16:02:26 +0000 |
commit | 3fec1ea7e35420ef87de8a98aed0437570251969 (patch) | |
tree | fbe033328d6e067fc209580928d37e53f1849657 /src/armnnSerializer/Serializer.hpp | |
parent | ffab16f9f23868b5b9338a1b63393f3be53211cb (diff) | |
download | armnn-3fec1ea7e35420ef87de8a98aed0437570251969.tar.gz |
IVGCVSW-4067 Change LayerGuid to use ProfilingGuid
* Refactoring to enable ProfilingGuid
* Add profiling includes to Android.mk
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: Ieb25e15e3dc302eb42817d824ad8411ac76dcfe8
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r-- | src/armnnSerializer/Serializer.hpp | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index 1fd507af80..7dfd534081 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -13,6 +13,8 @@ #include <ArmnnSchema_generated.h> +#include <armnn/Types.hpp> + namespace armnnSerializer { @@ -27,12 +29,12 @@ public: return m_flatBufferBuilder; } - std::vector<unsigned int>& GetInputIds() + std::vector<uint32_t>& GetInputIds() { return m_inputIds; } - std::vector<unsigned int>& GetOutputIds() + std::vector<uint32_t>& GetOutputIds() { return m_outputIds; } @@ -277,7 +279,7 @@ private: flatbuffers::Offset<flatbuffers::Vector<T>> CreateDataVector(const void* memory, unsigned int size); ///Function which maps Guid to an index - uint32_t GetSerializedId(unsigned int guid); + uint32_t GetSerializedId(armnn::LayerGuid guid); /// Creates the serializer InputSlots for the layer. std::vector<flatbuffers::Offset<armnnSerializer::InputSlot>> CreateInputSlots( @@ -293,14 +295,14 @@ private: /// AnyLayers required by the SerializedGraph. std::vector<flatbuffers::Offset<armnnSerializer::AnyLayer>> m_serializedLayers; - /// Guids of all Input Layers required by the SerializedGraph. - std::vector<unsigned int> m_inputIds; + /// Vector of indexes of all Input Layers required by the SerializedGraph. + std::vector<uint32_t> m_inputIds; - /// Guids of all Output Layers required by the SerializedGraph. - std::vector<unsigned int> m_outputIds; + /// Vector of indexes of all Output Layers required by the SerializedGraph. + std::vector<uint32_t> m_outputIds; /// Mapped Guids of all Layers to match our index. - std::unordered_map<unsigned int, uint32_t > m_guidMap; + std::unordered_map<armnn::LayerGuid, uint32_t > m_guidMap; /// layer within our FlatBuffer index. uint32_t m_layerId; |