diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-09 15:56:37 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-09 16:00:45 -0800 |
| commit | 605a0b8528615fa0b48f5d9c93eb8cc4a9006789 (patch) | |
| tree | 9e95928a09437c35b728d85051b18c2bb05846ad | |
| parent | d6326d772d117e6950304c7240cb029642fbf58c (diff) | |
[tpu:profiler] Capture the data for generating an overview page of the profiling results.
PiperOrigin-RevId: 181387984
| -rw-r--r-- | tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc b/tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc index bff23a447f..b67f2f47a7 100644 --- 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; |