From 2be68ed6578b64929fde25bc45bdc85f79e4ef53 Mon Sep 17 00:00:00 2001 From: Inki Dae Date: Wed, 9 Dec 2020 15:36:09 +0900 Subject: [PATCH] Fix backend name when profiling This patch corrects a backend name when profiling via MLAPI. In case of MLAPI backend, there are two cases. One is "ONE", and other is "TFLITE" in default. So this patch changes actual backend name by checking backend_type in runtime when logging the profile data, backend name. Change-Id: I0126cd6d427aee760df6224b2813b532d584d287 Signed-off-by: Inki Dae --- src/inference_engine_common_impl.cpp | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/inference_engine_common_impl.cpp b/src/inference_engine_common_impl.cpp index f65c5f3..a39cb17 100644 --- a/src/inference_engine_common_impl.cpp +++ b/src/inference_engine_common_impl.cpp @@ -341,7 +341,12 @@ out: } if (mUseProfiler == true) { - mProfiler.AddBackendName(config->backend_name); + if (config->backend_type == INFERENCE_BACKEND_ONE) { + std::string backend_name = "one"; + mProfiler.AddBackendName(backend_name); + } else { + mProfiler.AddBackendName(config->backend_name); + } } LOGI("LEAVE"); -- 2.34.1