diff options
| author |  Andrew Harp <andrewharp@google.com> | 2017-03-27 12:54:57 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-27 14:04:26 -0700 |
| commit | 4c46436cb52a536db258352b0384392a06e56a6d (patch) | |
| tree | 777f9488a126e9c22bffe8b1858dfd59c4f3a8cd /tensorflow/core/util/stat_summarizer.h | |
| parent | 94f610c1f2861744d62820153dd10839458978e3 (diff) | |
Output per-node-type timings to proto in benchmark_model
Change: 151366175
Diffstat (limited to 'tensorflow/core/util/stat_summarizer.h')
| -rw-r--r-- | tensorflow/core/util/stat_summarizer.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/util/stat_summarizer.h b/tensorflow/core/util/stat_summarizer.h index ea12111a39..23ba58a384 100644 --- a/tensorflow/core/util/stat_summarizer.h +++ b/tensorflow/core/util/stat_summarizer.h @@ -169,6 +169,11 @@ class StatSummarizer { // Prints the output tensor sizes and types for each node. void PrintOutputs() const; + void ComputeStatsByType(std::map<string, int64>* node_type_map_count, + std::map<string, int64>* node_type_map_time, + std::map<string, int64>* node_type_map_memory, + int64* accumulated_us) const; + std::string GetStatsByNodeType() const; std::string GetStatsByMetric(const string& title, |