Add detailed breakdow for stall duration in the analysis.

PiperOrigin-RevId: 561768588

plugin/tensorboard_plugin_profile/protobuf/dcn_slack_analysis.proto

@@ -2,6 +2,11 @@ syntax = "proto3";
 
 package tensorboard_plugin_profile;
 
+message OpInstance {
+  uint64 start_time_ns = 1;
+  uint64 duration_ns = 2;
+}
+
 message DcnSlack {
   string rendezvous = 1;
   // Xprof observed send start time.
@@ -39,6 +44,14 @@ message DcnSlack {
 
   // Send op name
   string send_op_name = 8;
+
+  // Timestamp for the send/send-done/recv/recv-done ops
+  OpInstance send = 9;
+  OpInstance send_done = 10;
+  OpInstance recv = 11;
+  OpInstance recv_done = 12;
+
+  string transfer_type = 13;
 }
 
 message DcnSlackSummary {
@@ -59,6 +72,14 @@ message DcnSlackSummary {
 
   // Send op name.
   string send_op_name = 8;
+
+  // Stall duration based on the op.
+  uint64 send_duration_us = 9;
+  uint64 recv_duration_us = 10;
+  uint64 send_done_duration_us = 11;
+  uint64 recv_done_duration_us = 12;
+
+  string transfer_type = 13;
 }
 
 message DcnSlackAnalysis {