for (const FunctionDef& function : graph.library().function()) {
    for (const NodeDef& node : function.node_def()) {
      if (NodeCanTriggerXlaCompilation(node)) {
        return true;
      }
    }
  }

  for (const NodeDef& node : graph.node()) {
    if (NodeCanTriggerXlaCompilation(node)) {
      return true;
    }
  }

  return false;
}