// ops, require replica id to be added as an op attribute to be used during
// execution. Handle such ops separately and add an integer attribute that
// represents replica id.
LogicalResult UpdateRegionReplicateVariantOps(
    OpBuilder& builder, Location loc, Region& region, int replica_id,
    const std::optional<DictionaryAttr>& devices) {
  auto result = region.walk([&](Operation* op) -> WalkResult {