Issue 65632
Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this [failed invocation](https://github.com/tensorflow/tensorflow/pull/66308/checks?check_run_id=24164887942) of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request. This PR is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you. Hi @bmoroch Can you please check the above [comment](https://github.com/tensorflow/tensorflow/pull/66308#pullrequestreview-2051218939) from @mihaimaruseac and keep us posted? Thank you!
Added fix for issue #65632: Check failed in tf.strings.unsorted_segment_join when the rank of inputs is larger than 1 and the value of num_segments is too large.