}

    attrs.emplace_back(builder.getNamedAttr(
        "decomposition",
        mlir::vhlo::StringV1Attr::get(builder.getContext(), decomposition)));
  }

  op_state.addAttributes(attrs);

  // Handle the conversion from subgraph index to functions for If and While. We
  // will add CallOps in the region to call the functions later for While.
  TF_ASSIGN_OR_RETURN(auto function_ref_attrs,