From db65d28dbb59890f945054c59d33cff0d5c3f4fe Mon Sep 17 00:00:00 2001 From: Chao Mei Date: Wed, 15 Apr 2020 00:02:31 -0700 Subject: [PATCH] Add a log when a tflite delegate is created. PiperOrigin-RevId: 306590390 Change-Id: I98b5d17b95747420b2dd0c3c0c037a6088cd495e --- tensorflow/lite/tools/evaluation/BUILD | 1 + .../lite/tools/evaluation/evaluation_delegate_provider.cc | 2 ++ 2 files changed, 3 insertions(+) diff --git a/tensorflow/lite/tools/evaluation/BUILD b/tensorflow/lite/tools/evaluation/BUILD index bf21e553b1f..1b74de33885 100644 --- a/tensorflow/lite/tools/evaluation/BUILD +++ b/tensorflow/lite/tools/evaluation/BUILD @@ -71,6 +71,7 @@ cc_library( deps = [ ":utils", "//tensorflow/lite/tools:command_line_flags", + "//tensorflow/lite/tools:logging", "//tensorflow/lite/tools:tool_params", "//tensorflow/lite/tools/benchmark:delegate_provider_hdr", "//tensorflow/lite/tools/benchmark:tflite_execution_providers", diff --git a/tensorflow/lite/tools/evaluation/evaluation_delegate_provider.cc b/tensorflow/lite/tools/evaluation/evaluation_delegate_provider.cc index 91a0aea4711..e7fc8296d8b 100644 --- a/tensorflow/lite/tools/evaluation/evaluation_delegate_provider.cc +++ b/tensorflow/lite/tools/evaluation/evaluation_delegate_provider.cc @@ -16,6 +16,7 @@ limitations under the License. #include "tensorflow/lite/tools/evaluation/evaluation_delegate_provider.h" #include "tensorflow/lite/tools/command_line_flags.h" +#include "tensorflow/lite/tools/logging.h" namespace tflite { namespace evaluation { @@ -118,6 +119,7 @@ std::vector DelegateProviders::CreateAllDelegates( // user-specified benchmark params tells not to. if (ptr == nullptr) continue; delegates.emplace_back(std::move(ptr)); + TFLITE_LOG(INFO) << one->GetName() << " delegate is created."; } return delegates; }