aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/benchmark
diff options
context:
space:
mode:
authorGravatar Asim Shankar <ashankar@google.com>2017-01-30 21:09:16 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-01-30 21:34:48 -0800
commit8691453b900aef43bf4a3966a5a96b14fbb7bfc3 (patch)
tree50f653d76cda5ca6d1daaa258ed39f52bcd5c571 /tensorflow/tools/benchmark
parent643fc7712fcedd1252eae855e4e89fc642cf0426 (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.cc3
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);