diff options
Diffstat (limited to 'tensorflow/core/util/stat_summarizer.h')
| -rw-r--r-- | tensorflow/core/util/stat_summarizer.h | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/core/util/stat_summarizer.h b/tensorflow/core/util/stat_summarizer.h index 7fc43d449b..57b17bd208 100644 --- a/tensorflow/core/util/stat_summarizer.h +++ b/tensorflow/core/util/stat_summarizer.h @@ -16,14 +16,14 @@ limitations under the License. #ifndef TENSORFLOW_UTIL_STAT_SUMMARIZER_H_ #define TENSORFLOW_UTIL_STAT_SUMMARIZER_H_ +#include <stdlib.h> + #include <cmath> #include <limits> #include <map> #include <sstream> #include <string> -#include <stdlib.h> - #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/platform/types.h" @@ -158,6 +158,8 @@ class StatSummarizer { double cdf_cutoff_ratio = 1.0, int num_max_nodes_to_print = std::numeric_limits<int>::max()) const; + std::string GetStatsByNodeType() const; + void Reset() { run_total_micros_.Reset(); memory_.Reset(); @@ -184,6 +186,8 @@ class StatSummarizer { BY_RUN_ORDER, }; + void Validate(const Detail* detail, const NodeExecStats& ns) const; + // Summarizes all nodes' stat in the order of node names defined in the graph. std::string GetStatsByOrderOfNodeDefinitions(bool use_memory) const; |