From a04a9d7c11f28c7e932435535e80223782f369f2 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Tue, 27 Apr 2021 10:02:10 +0100 Subject: IVGCVSW-5775 'Add Async Support to ExecuteNetwork' * Enabled async mode with '-n, concurrent' and 'simultaneous-iterations' in ExecuteNetwork * Number of input files provided should be equal to number of input files provided multiply by number of simultaneous iterations divided by comma !armnn:5443 Signed-off-by: Sadik Armagan Change-Id: Ibeb318010430bf4ae61a02b18b1bf88f3657774c --- .../reference/workloads/RefDetectionPostProcessWorkload.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/backends/reference/workloads/RefDetectionPostProcessWorkload.cpp') diff --git a/src/backends/reference/workloads/RefDetectionPostProcessWorkload.cpp b/src/backends/reference/workloads/RefDetectionPostProcessWorkload.cpp index 25c326ad37..6784e21585 100644 --- a/src/backends/reference/workloads/RefDetectionPostProcessWorkload.cpp +++ b/src/backends/reference/workloads/RefDetectionPostProcessWorkload.cpp @@ -46,10 +46,10 @@ void RefDetectionPostProcessWorkload::Execute(std::vector inputs auto scores = MakeDecoder(scoresInfo, inputs[1]->Map()); auto anchors = MakeDecoder(anchorsInfo, m_Anchors->Map(false)); - float* detectionBoxes = GetOutputTensorData(0, m_Data); - float* detectionClasses = GetOutputTensorData(1, m_Data); - float* detectionScores = GetOutputTensorData(2, m_Data); - float* numDetections = GetOutputTensorData(3, m_Data); + float* detectionBoxes = reinterpret_cast(outputs[0]->Map()); + float* detectionClasses = reinterpret_cast(outputs[1]->Map()); + float* detectionScores = reinterpret_cast(outputs[2]->Map()); + float* numDetections = reinterpret_cast(outputs[3]->Map()); DetectionPostProcess(boxEncodingsInfo, scoresInfo, anchorsInfo, detectionBoxesInfo, detectionClassesInfo, -- cgit v1.2.1