From 27a9bd93bd0f02c8ad866ae0fb70f1c4bb67ef0c Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Thu, 12 Nov 2020 15:48:34 +0000 Subject: This application allow to benchmark tflite models by providing average inference time. Usage: armnn_tfl_benchmark -m -m --model_file <.tflite file path>: .tflite model to be executed -b --backend : preferred backend device to run layers on by default. Possible choices: CpuAcc, CpuRef -l --loops : provide the number of time the inference will be executed (by default nb_loops=1) Signed-off-by: Vincent ABRIOU Signed-off-by: Jim Flynn Change-Id: Ia26fafd4f382f0ad03856436dcae6e71b5abbd26 --- tests/CMakeLists.txt | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'tests/CMakeLists.txt') diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index edea34dfee..135f6497be 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -137,6 +137,10 @@ if (BUILD_TF_LITE_PARSER) addDllCopyCommands(${testName}) endmacro() + set(TfLiteBenchmark-Armnn_sources + TfLiteBenchmark-Armnn/TfLiteBenchmark-Armnn.cpp) + TfLiteParserTest(TfLiteBenchmark-Armnn "${TfLiteBenchmark-Armnn_sources}") + set(TfLiteMobilenetQuantized-Armnn_sources TfLiteMobilenetQuantized-Armnn/TfLiteMobilenetQuantized-Armnn.cpp ImagePreprocessor.hpp -- cgit v1.2.1