From 7478ab55d6148f951a90afff11111b069f76af95 Mon Sep 17 00:00:00 2001
From: Keith Mok <ek9852@gmail.com>
Date: Tue, 2 Mar 2021 10:07:59 -0800
Subject: 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 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp b/tests/TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp
index a010717412..4e092cf85d 100644
--- a/tests/TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp
+++ b/tests/TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp
@@ -191,7 +191,7 @@ int main(int argc, char* argv[])
     {
         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
-- 
cgit v1.2.1