diff options
author | 2018-09-04 12:22:27 -0700 | |
---|---|---|
committer | 2018-09-04 12:26:46 -0700 | |
commit | f9b58d46499c79e01f55d9e16867a8aace667db8 (patch) | |
tree | b15598feb0ac6f237c63f6e1ff1ed1a35e19c8bc /tensorflow/contrib/tpu | |
parent | 5587f4eb011115b947daaa4b092ef70650705687 (diff) |
Add more data fields to step proto.
PiperOrigin-RevId: 211501909
Diffstat (limited to 'tensorflow/contrib/tpu')
-rw-r--r-- | tensorflow/contrib/tpu/profiler/tf_op_stats.proto | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/contrib/tpu/profiler/tf_op_stats.proto b/tensorflow/contrib/tpu/profiler/tf_op_stats.proto index 2b13343efa..f88dc51636 100644 --- a/tensorflow/contrib/tpu/profiler/tf_op_stats.proto +++ b/tensorflow/contrib/tpu/profiler/tf_op_stats.proto @@ -79,12 +79,15 @@ message StepInfoResult { // The step duration in picoseconds. optional uint64 duration_ps = 2; // The infeed duration in picoseconds. - // Can turn into a map if we want a variable number of ops. optional uint64 infeed_duration_ps = 3; + // The outfeed duration in picoseconds. + optional uint64 host_outfeed_ps = 8; // The start time of this step in picoseconds. optional uint64 begin_ps = 4; // The waiting time within this step in picoseconds. optional uint64 wait_duration_ps = 5; + // The unit b outfeed duration in picoseconds. + optional uint64 unit_b_outfeed_ps = 9; // The time spent on cross-replica-sum in picoseconds. optional uint64 crs_duration_ps = 6; // Percentage of unit b time spent on infeed. |