From 9fc9d5bfc460f736befa25f640a8010664945d61 Mon Sep 17 00:00:00 2001
From: Shashi Shekhar <shashishekhar@google.com>
Date: Wed, 23 May 2018 18:45:30 -0700
Subject: [PATCH] Add back some public interface methods.

PiperOrigin-RevId: 197826136
---
 tensorflow/core/util/stat_summarizer.h | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tensorflow/core/util/stat_summarizer.h b/tensorflow/core/util/stat_summarizer.h
index 39cd948..173ed5c 100644
--- a/tensorflow/core/util/stat_summarizer.h
+++ b/tensorflow/core/util/stat_summarizer.h
@@ -95,6 +95,13 @@ class StatSummarizer {
                                                num_stats);
   }
 
+  int num_runs() const { return stats_calculator_->num_runs(); }
+
+  // Returns stats of total microseconds spent by all nodes in each run.
+  const Stat<int64_t>& run_total_us() const {
+    return stats_calculator_->run_total_us();
+  }
+
  private:
   void Validate(const std::vector<TensorDescription>* outputs,
                 const NodeExecStats& ns) const;
-- 
2.7.4