kImportModelDefaultGraphFuncName);
      main_func_op != nullptr && IsPublicFuncOp(main_func_op)) {
    return main_func_op;
  }

  if (const auto serving_default_func_op =
          module_op.lookupSymbol<func::FuncOp>(kDefaultServingSignatureDefKey);
      serving_default_func_op != nullptr &&
      IsPublicFuncOp(serving_default_func_op)) {
    return serving_default_func_op;
  }

  return nullptr;
}