diff options
author | Cathal Corbett <cathal.corbett@arm.com> | 2022-03-04 10:36:34 +0000 |
---|---|---|
committer | Cathal Corbett <cathal.corbett@arm.com> | 2022-03-07 11:19:08 +0000 |
commit | 19793551a67a5afd4cfb68e5dda7751e3d321a35 (patch) | |
tree | 355707246183ea982d4e6fead5be25f038111624 /src/armnn/LoadedNetwork.cpp | |
parent | 344302581b66677a748a456f370752db75adde21 (diff) | |
download | armnn-19793551a67a5afd4cfb68e5dda7751e3d321a35.tar.gz |
IVGCVSW-6817 Add IProfilingService& as an argument to the IReportStructure
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ib8e75eea49debe3b1dd8fa72623a55b26cb6ded4
Diffstat (limited to 'src/armnn/LoadedNetwork.cpp')
-rw-r--r-- | src/armnn/LoadedNetwork.cpp | 26 |
1 files changed, 13 insertions, 13 deletions
diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index 1dbd1e3112..46c1ce58aa 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -543,14 +543,14 @@ void LoadedNetwork::AllocateAndExecuteConstantWorkloadsAsync() } } -void LoadedNetwork::SendNetworkStructure() +void LoadedNetwork::SendNetworkStructure(arm::pipe::IProfilingService& profilingService) { ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "LoadNetwork_SendNetworkStructure"); Graph& order = m_OptimizedNetwork->pOptimizedNetworkImpl->GetGraph().TopologicalSort(); ProfilingGuid networkGuid = m_OptimizedNetwork->GetGuid(); std::unique_ptr<TimelineUtilityMethods> timelineUtils = - TimelineUtilityMethods::GetTimelineUtils(*m_ProfilingService); + TimelineUtilityMethods::GetTimelineUtils(profilingService); timelineUtils->CreateTypedEntity(networkGuid, LabelsAndEventClasses::NETWORK_GUID); @@ -560,19 +560,19 @@ void LoadedNetwork::SendNetworkStructure() AddLayerStructure(timelineUtils, *layer, networkGuid); switch (layer->GetType()) { - case LayerType::Input: - case LayerType::Output: - { - // Inputs and outputs are treated in a special way - see EnqueueInput() and EnqueueOutput(). - break; - } - default: + case LayerType::Input: + case LayerType::Output: { - for (auto& workload : m_WorkloadQueue) - { - // Add workload to the post-optimisation network structure - AddWorkloadStructure(timelineUtils, workload, *layer); + // Inputs and outputs are treated in a special way - see EnqueueInput() and EnqueueOutput(). + break; } + default: + { + for (auto& workload : m_WorkloadQueue) + { + // Add workload to the post-optimisation network structure + AddWorkloadStructure(timelineUtils, workload, *layer); + } break; } } |