From b2b13549d763b75e160744bc00ee1c526b419c03 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 25 May 2018 13:39:25 -0700 Subject: [PATCH] [tpu:profiler] Capture the data for generating a memory viewer of the profiling results. PiperOrigin-RevId: 198095564 --- tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc b/tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc index e51f645..9948532 100644 --- a/tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc +++ b/tensorflow/contrib/tpu/profiler/capture_tpu_profile.cc @@ -81,6 +81,7 @@ ProfileRequest PopulateProfileRequest(int duration_ms, } request.add_tools("op_profile"); request.add_tools("input_pipeline"); + request.add_tools("memory_viewer"); request.add_tools("overview_page"); *request.mutable_opts() = opts; std::cout << "Limiting the number of trace events to " << kMaxEvents -- 2.7.4