diff options
author | 2018-06-04 15:00:11 -0700 | |
---|---|---|
committer | 2018-06-04 15:02:44 -0700 | |
commit | e2d300823f410823b1b5fee4e5159a754247e219 (patch) | |
tree | c7f31757883d8689f0a24a58e8d64e3b42f36058 /tensorflow/core/util/stat_summarizer.cc | |
parent | 142ccf3666e07d011aa83fdd6be8c17f721fbc99 (diff) |
Move benchmarking code to a new directory and add some documentation.
PiperOrigin-RevId: 199200246
Diffstat (limited to 'tensorflow/core/util/stat_summarizer.cc')
-rw-r--r-- | tensorflow/core/util/stat_summarizer.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/core/util/stat_summarizer.cc b/tensorflow/core/util/stat_summarizer.cc index 42a4801dcb..a5c1fda102 100644 --- a/tensorflow/core/util/stat_summarizer.cc +++ b/tensorflow/core/util/stat_summarizer.cc @@ -78,6 +78,14 @@ void StatSummarizer::Validate(const std::vector<TensorDescription>* outputs, } } +void StatSummarizer::PrintStepStats() const { + string output = GetOutputString(); + std::istringstream iss(output); + for (std::string line; std::getline(iss, line);) { + LOG(INFO) << line; + } +} + namespace { std::string OpType(const DeviceStepStats& ds, const NodeExecStats& ns) { // There is no published specification of how DeviceStats and NodeStats |