[XLA:Python] Fix build failure on Mac OS.
[XLA:Python] Fix build failure on Mac OS.

ssize_t and int64_t aren't the same thing on Mac. But we always want to use int64_t, so just change the type to that and convert around the NumPy call.
