diff options
| -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 |