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/armnn | |
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/armnn')
-rw-r--r-- | src/armnn/Layer.cpp | 12 | ||||
-rw-r--r-- | src/armnn/QuantizerVisitor.cpp | 4 |
2 files changed, 4 insertions, 12 deletions
diff --git a/src/armnn/Layer.cpp b/src/armnn/Layer.cpp index dbeda22ca0..1efe7e412f 100644 --- a/src/armnn/Layer.cpp +++ b/src/armnn/Layer.cpp @@ -5,6 +5,7 @@ #include "Layer.hpp" #include "Graph.hpp" +#include <ProfilingService.hpp> #include <backendsCommon/WorkloadData.hpp> #include <backendsCommon/CpuTensorHandle.hpp> @@ -184,15 +185,6 @@ EdgeStrategy OutputSlot::GetEdgeStrategyForConnection(unsigned int connectionIdx return m_EdgeStrategies[connectionIdx]; } -namespace { -LayerGuid GenerateLayerGuid() -{ - // Note: Not thread safe. - static LayerGuid newGuid=0; - return newGuid++; -} -} // namespace - Layer::Layer(unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, @@ -202,7 +194,7 @@ Layer::Layer(unsigned int numInputSlots, , m_LayerName(name ? name : "") , m_Type(type) , m_BackendId() -, m_Guid(GenerateLayerGuid()) +, m_Guid(profiling::ProfilingService::Instance().NextGuid()) { m_InputSlots.reserve(numInputSlots); for (unsigned int i = 0; i < numInputSlots; ++i) diff --git a/src/armnn/QuantizerVisitor.cpp b/src/armnn/QuantizerVisitor.cpp index 8a3d4f2990..4b80b02e34 100644 --- a/src/armnn/QuantizerVisitor.cpp +++ b/src/armnn/QuantizerVisitor.cpp @@ -109,8 +109,8 @@ ConstTensor QuantizerVisitor::CreateQuantizedBias(const IConnectableLayer* srcLa void QuantizerVisitor::RecordLayer(const IConnectableLayer* srcLayer, IConnectableLayer* quantizedLayer) { - m_OriginalToQuantizedGuidMap[srcLayer->GetGuid()] = quantizedLayer->GetGuid(); - m_QuantizedGuidToLayerMap[quantizedLayer->GetGuid()] = quantizedLayer; + m_OriginalToQuantizedGuidMap.insert(std::make_pair(srcLayer->GetGuid(), quantizedLayer->GetGuid())); + m_QuantizedGuidToLayerMap.insert(std::make_pair(quantizedLayer->GetGuid(), quantizedLayer)); } void QuantizerVisitor::VisitAbsLayer(const IConnectableLayer* layer, const char* name) |