diff options
Diffstat (limited to 'tensorflow/tools/benchmark')
| -rw-r--r-- | tensorflow/tools/benchmark/benchmark_model.cc | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/tools/benchmark/benchmark_model.cc b/tensorflow/tools/benchmark/benchmark_model.cc index 86416db953..180600e3b4 100644 --- a/tensorflow/tools/benchmark/benchmark_model.cc +++ b/tensorflow/tools/benchmark/benchmark_model.cc @@ -370,8 +370,7 @@ int Main(int argc, char** argv) { stats_options.memory_limit = memory_limit; stats_options.show_type = show_type; stats_options.show_summary = show_summary; - stats.reset( - new tensorflow::StatSummarizer(*(graph_def.get()), stats_options)); + stats.reset(new tensorflow::StatSummarizer(stats_options)); const double sleep_seconds = std::strtod(run_delay.c_str(), nullptr); |