commit 7478ab55d6148f951a90afff11111b069f76af95
author Keith Mok <ek9852@gmail.com> Tue Mar 02 10:07:59 2021 -0800
committer mike.kelly <mike.kelly@arm.com> Wed Mar 03 09:21:56 2021 +0000
tree f5b19df5ba716449a54e79f88c804c09fc65bc00
parent df6ed10236147a38f009f7be65db870c2de7d34b

armnn_tfl_benchmark: Fix crash

The input tensor data pointer was passed incorrectly

Signed-off-by: Keith Mok <ek9852@gmail.com>
Change-Id: I03f872c57ec588fde0f7d444c80b38823ea4f9b4

tests/TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp

diff --git a/tests/TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp b/tests/TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp
index a010717..4e092cf 100644
--- a/tests/TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp
+++ b/tests/TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp
@@ -191,7 +191,7 @@
     {
         std::vector<float> in_data(inputTensorInfos.at(i).GetNumElements());
         in.push_back(in_data);
-        inputTensors.push_back({ inputBindings[i].first, armnn::ConstTensor(inputBindings[i].second, in.data()) });
+        inputTensors.push_back({ inputBindings[i].first, armnn::ConstTensor(inputBindings[i].second, in[i].data()) });
     }
 
     // Allocate output tensors