Add DutyCycleAnalysis for TPUs to OpStats.

Fix DutyCycleTracker bugs with idleness and duplicate active times.

PiperOrigin-RevId: 714257416

xla/tsl/profiler/convert/xla_op_utils.h

@@ -129,6 +129,20 @@ inline bool MayHaveInnerOps(absl::string_view category) {
          category == kHloWhile || category == kHloMegacoreFusion;
 }
 
+inline bool IsOffDutyOp(absl::string_view category) {
+  return (category == tsl::profiler::kHloInfeed ||
+          category == tsl::profiler::kHloOutfeed ||
+          category == tsl::profiler::kHloHostSend ||
+          category == tsl::profiler::kHloHostSendDone ||
+          category == tsl::profiler::kHloHostRecv ||
+          category == tsl::profiler::kHloHostRecvDone ||
+          category ==
+              tsl::profiler::kHloMegacoreFusion  // Only self-time in megacore
+                                                 // fusion is off-duty. The op
+                                                 // time of children is on-duty.
+  );
+}
+
 }  // namespace profiler
 }  // namespace tsl