aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefDetectionPostProcessWorkload.cpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2021-04-27 10:02:10 +0100
committerSadik Armagan <sadik.armagan@arm.com>2021-04-29 08:46:09 +0000
commita04a9d7c11f28c7e932435535e80223782f369f2 (patch)
tree9c1e86b0b4878dad12a359e60a8d2e8e051d2def /src/backends/reference/workloads/RefDetectionPostProcessWorkload.cpp
parent484d5ebb00c0db76efd76a601b5bbaa460cd2ccb (diff)
downloadarmnn-a04a9d7c11f28c7e932435535e80223782f369f2.tar.gz
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 <sadik.armagan@arm.com> Change-Id: Ibeb318010430bf4ae61a02b18b1bf88f3657774c
Diffstat (limited to 'src/backends/reference/workloads/RefDetectionPostProcessWorkload.cpp')
-rw-r--r--src/backends/reference/workloads/RefDetectionPostProcessWorkload.cpp8
1 files changed, 4 insertions, 4 deletions
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<ITensorHandle*> inputs
auto scores = MakeDecoder<float>(scoresInfo, inputs[1]->Map());
auto anchors = MakeDecoder<float>(anchorsInfo, m_Anchors->Map(false));
- float* detectionBoxes = GetOutputTensorData<float>(0, m_Data);
- float* detectionClasses = GetOutputTensorData<float>(1, m_Data);
- float* detectionScores = GetOutputTensorData<float>(2, m_Data);
- float* numDetections = GetOutputTensorData<float>(3, m_Data);
+ float* detectionBoxes = reinterpret_cast<float*>(outputs[0]->Map());
+ float* detectionClasses = reinterpret_cast<float*>(outputs[1]->Map());
+ float* detectionScores = reinterpret_cast<float*>(outputs[2]->Map());
+ float* numDetections = reinterpret_cast<float*>(outputs[3]->Map());
DetectionPostProcess(boxEncodingsInfo, scoresInfo, anchorsInfo,
detectionBoxesInfo, detectionClassesInfo,