xla

218f4e54 - [Pallas] Make _histogram output int32 (#7202)

[Pallas] Make _histogram output int32 (#7202) Summary: Make _histogram output int32 such that it's compatible with TPUs. Test Plan: python test/test_gmm.py -v -k test_histogram