config.unlegalizedOps = &unconverted_ops;
  auto result = applyPartialConversion(op, target, std::move(patterns), config);
  if (failed(result)) {
    IncrementFailedLegalizationCount(op, target);
  }
  for (const auto &unconverted_op : unconverted_ops) {
    IncrementFailedLegalizationCount(unconverted_op, target);
  }
  return result;
}

/// When `tf2xla_fallback_device_type` is not `None`, also uses legalization