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

Adjust the order of parsing commandline options to avoid the confusing message about the --perf_options_list flag.

Browse Source 
PiperOrigin-RevId: 333711021
Change-Id: I64cc6c867baa608718f08f88884712d241ea18c2
This commit is contained in:
Chao Mei 2020-09-25 04:55:03 -07:00 committed by TensorFlower Gardener
parent 9aff666a8d
commit d6f2f98875
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/lite/tools/benchmark/benchmark_performance_options.cc
View File

@ -360,13 +360,13 @@ void BenchmarkPerformanceOptions::Run() {
} }
void BenchmarkPerformanceOptions::Run(int argc, char** argv) { void BenchmarkPerformanceOptions::Run(int argc, char** argv) {
// We first parse flags for single-option runs to get information like // Parse flags that are supported by this particular binary first.
// parameters of the input model etc.
if (single_option_run_->ParseFlags(&argc, argv) != kTfLiteOk) return;
// Now, we parse flags that are specified for this particular binary.
if (!ParseFlags(&argc, argv)) return; if (!ParseFlags(&argc, argv)) return;
// Then parse flags for single-option runs to get information like parameters
// of the input model etc.
if (single_option_run_->ParseFlags(&argc, argv) != kTfLiteOk) return;
// Now, the remaining are unrecognized flags and we simply print them out. // Now, the remaining are unrecognized flags and we simply print them out.
for (int i = 1; i < argc; ++i) { for (int i = 1; i < argc; ++i) {
TFLITE_LOG(WARN) << "WARNING: unrecognized commandline flag: " << argv[i]; TFLITE_LOG(WARN) << "WARNING: unrecognized commandline flag: " << argv[i];