segment_reduction_ops_gpu: Fix MSVC compile errors
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/61309/checks?check_run_id=15122111588) 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. Hi @johnnkp Can you please sign CLA. Thank you! > Hi @johnnkp Can you please sign CLA. Thank you!

Hi @johnnkp CLA looks good now. Thank you! https://github.com/tensorflow/tensorflow/issues/61310 is related to this PR. Some changes may be unnecessary under `clang-cl`.
This is a draft PR to fix https://github.com/tensorflow/tensorflow/issues/60397. More Windows specific macros may be necessary to be added.