experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add a log when a tflite delegate is created.

Browse Source 
PiperOrigin-RevId: 306590390
Change-Id: I98b5d17b95747420b2dd0c3c0c037a6088cd495e
This commit is contained in:
Chao Mei 2020-04-15 00:02:31 -07:00 committed by TensorFlower Gardener
parent 2ddbc3572c
commit db65d28dbb
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/lite/tools/evaluation/BUILD
View File

@ -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",

2
tensorflow/lite/tools/evaluation/evaluation_delegate_provider.cc
View File

@ -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<TfLiteDelegatePtr> 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;
}