Fixing build issue with Clang 16
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/61503/checks?check_run_id=15728974098) 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 @matthew-olson-intel Can you please sign CLA. Thank you! @gbaned Done!
This is a PR which fixes the TensorFlow build when using Clang 16; this file previously used `uint64_t` without including the proper header.

#61289 