message: [XLA] Correctly propagate builder and partition_id for inner_dot_builder.
  Fixes crash when some order of partitioning is triggered.
diff: @@ -2237,7 +2237,10 @@ StatusOr<HloInstruction*> PartitionDotGroupOnContracting(
  }
  // Use resharding to slice the output. Use a temporary reshard cache since
  // we are faking with replicated sharding.
  - auto new_state = lhs.state();
  + PartitionedHlo::PartitioningState new_state = lhs.state();
  + new_state.b = b;
  + new_state.partition_id =
  + lhs.state().collective_ops_creator.create_partition_id(b);
  PartitionedHlo::ReshardCache tmp_cache;
  new_state.reshard_cache = &tmp_cache;
  ar->set_sharding(HloSharding::Replicate());
  @@ -2246,8 +2249,9 @@ StatusOr<HloInstruction*> PartitionDotGroupOnContracting(
  output_sharding, get_non_slice_dims()))
  .hlo();
  };
  - auto inner_state = CreatePerGroupPartitioningState(
  - lhs.state(), lhs_grouped.device_groups, b);
  + PartitionedHlo::PartitioningState inner_state =
  + CreatePerGroupPartitioningState(lhs.state(), lhs_grouped.device_groups,
  + b);
  TF_ASSIGN_OR_RETURN(
  auto dot,
  PartitionDot(
  