diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-05-03 08:52:26 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-03 10:03:55 -0700 |
commit | 00bfa9069e2f8456e26e09d6a7a1d0dff807e5bf (patch) | |
tree | c6c6ecbfaf391c08019f3bb77076bec0e4e7f81c /tensorflow/tools/benchmark | |
parent | b93dd62e8ae9faed909c677781edc278632704f0 (diff) |
fixed a bug passing two flag values: show_type and show_summary, they currently override the value of show_time
Change: 154967526
Diffstat (limited to 'tensorflow/tools/benchmark')
-rw-r--r-- | tensorflow/tools/benchmark/benchmark_model.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/tools/benchmark/benchmark_model.cc b/tensorflow/tools/benchmark/benchmark_model.cc index c2e41e4918..8c480f8d9d 100644 --- a/tensorflow/tools/benchmark/benchmark_model.cc +++ b/tensorflow/tools/benchmark/benchmark_model.cc @@ -334,8 +334,8 @@ int Main(int argc, char** argv) { Flag("show_memory", &show_memory, "whether to list stats by memory used"), Flag("memory_limit", &memory_limit, "how many items to show by memory used"), - Flag("show_type", &show_time, "whether to list stats by op type"), - Flag("show_summary", &show_time, + Flag("show_type", &show_type, "whether to list stats by op type"), + Flag("show_summary", &show_summary, "whether to show a summary of the stats"), Flag("show_flops", &show_flops, "whether to estimate the model's FLOPs"), Flag("warmup_runs", &warmup_runs, "how many runs to initialize model"), |