diff options
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, |