[tpu:profiler] Capture the data for generating an overview page of the profiling...

PiperOrigin-RevId: 181387984

diff --git a/tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc b/tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc
index bff23a447f841339d9bf5bd3bf125d705bf1fee7..b67f2f47a7b753fd4629d7ad4db0b4c67933ce0b 100644 (file)
--- a/tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc
+++ b/tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc
@@ -51,6 +51,7 @@ ProfileResponse Profile(const string& service_addr, int duration_ms) {
   request.set_duration_ms(duration_ms);
   request.set_max_events(kMaxEvents);
   request.add_tools("input_pipeline");
+  request.add_tools("overview_page");
   std::cout << "Limiting the number of trace events to " << kMaxEvents
             << std::endl;
   ::grpc::ClientContext context;