message: More accurate input-pipeline analysis for TPU.
diff: @@ -116,18 +116,18 @@ EventType ClassifyGpuEvent(absl::string_view event_name) {
  }
  EventType ClassifyCpuEvent(absl::string_view event_name, int64 correlation_id) {
  - if (absl::StartsWithIgnoreCase(event_name, "MEMCPYHtoD"))
  + if (absl::StartsWithIgnoreCase(event_name, "MEMCPYHtoD") ||
  + absl::StrContains(event_name, "Infeed"))
  return HOST_TO_DEVICE;
  if (absl::StartsWithIgnoreCase(event_name, "MEMCPYHtoH")) return HOST_TO_HOST;
  if (correlation_id >= 0 ||
  absl::StartsWithIgnoreCase(event_name, "ExecutorState::Process")) {
  return HOST_PREPARE;
  - } else {
  + }
  if (absl::StartsWithIgnoreCase(event_name, "IteratorGetNext"))
  return HOST_WAIT_INPUT;
  return HOST_COMPUTE;
  }
  -}
  std::string PrintEventType(EventType event_type) {
  switch (event_type) {
  