From: A. Unique TensorFlower Date: Fri, 25 May 2018 20:39:25 +0000 (-0700) Subject: [tpu:profiler] Capture the data for generating a memory viewer of the profiling... X-Git-Tag: upstream/v1.9.0_rc1~38^2~4^2~51 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=b2b13549d763b75e160744bc00ee1c526b419c03;p=platform%2Fupstream%2Ftensorflow.git [tpu:profiler] Capture the data for generating a memory viewer of the profiling results. PiperOrigin-RevId: 198095564 --- 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