diff options
author | Keith Mok <ek9852@gmail.com> | 2021-03-02 10:07:59 -0800 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2021-03-03 09:21:56 +0000 |
commit | 7478ab55d6148f951a90afff11111b069f76af95 (patch) | |
tree | f5b19df5ba716449a54e79f88c804c09fc65bc00 | |
parent | df6ed10236147a38f009f7be65db870c2de7d34b (diff) | |
download | armnn-7478ab55d6148f951a90afff11111b069f76af95.tar.gz |
armnn_tfl_benchmark: Fix crash
The input tensor data pointer was passed incorrectly
Signed-off-by: Keith Mok <ek9852@gmail.com>
Change-Id: I03f872c57ec588fde0f7d444c80b38823ea4f9b4
-rw-r--r-- | tests/TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
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 |