mlir::SymbolRefAttr::get(builder.getContext(), func_names.at(body_idx));
    auto cond_attr =
        mlir::SymbolRefAttr::get(builder.getContext(), func_names.at(cond_idx));

    op_state.addAttribute("body", body_attr);
    op_state.addAttribute("cond", cond_attr);

    return absl::OkStatus();
  }
  if (auto* opts = op.builtin_options_2.AsStablehloScatterOptions()) {