diff options
author | Asim Shankar <ashankar@google.com> | 2017-01-30 21:09:16 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-30 21:34:48 -0800 |
commit | 8691453b900aef43bf4a3966a5a96b14fbb7bfc3 (patch) | |
tree | 50f653d76cda5ca6d1daaa258ed39f52bcd5c571 /tensorflow/tools/benchmark | |
parent | 643fc7712fcedd1252eae855e4e89fc642cf0426 (diff) |
StatSummarizer: Make it work without needing the GraphDef.
This will allow the StatSummarizer to be instantiated and used
even when the GraphDef is not easily accessible. A consequence of
this is that the BY_DEFINITION_ORDER ordering of stats is no
longer available, but that was deemed acceptable for this change.
Other notables:
- Added a basic C++ unittest for stat_summarizer.
- Added some commentary about caveats about summaries over runs
that involve GPUs or partitioned graphs. These caveats existed
in the prior implementation as well.
Change: 146076563
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); |