diff options
author | 2017-05-24 11:00:29 -0700 | |
---|---|---|
committer | 2017-05-24 11:04:01 -0700 | |
commit | b509dbf12e6fe249cefb8db8ae9b4ad43723939f (patch) | |
tree | 9a934bebab153c29469c9c6a2882525aa37a50c6 /tensorflow/core/util/stat_summarizer.h | |
parent | e2d574556188989cf540154932ed9193706f20f7 (diff) |
Track statistics for looping graphs in benchmark, and add missing kernels
PiperOrigin-RevId: 157010139
Diffstat (limited to 'tensorflow/core/util/stat_summarizer.h')
-rw-r--r-- | tensorflow/core/util/stat_summarizer.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/util/stat_summarizer.h b/tensorflow/core/util/stat_summarizer.h index 23ba58a384..6111e276ea 100644 --- a/tensorflow/core/util/stat_summarizer.h +++ b/tensorflow/core/util/stat_summarizer.h @@ -172,6 +172,7 @@ class StatSummarizer { 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, + std::map<string, int64>* node_type_map_times_called, int64* accumulated_us) const; std::string GetStatsByNodeType() const; @@ -201,6 +202,7 @@ class StatSummarizer { Stat<int64> rel_end_us; Stat<int64> mem_used; std::vector<TensorDescription> outputs; + int64 times_called; }; void Validate(const Detail* detail, const NodeExecStats& ns) const; |