Switch int64 to int32 in a few places. TF 1.6 TPU support for int64 i…

…s spotty.

PiperOrigin-RevId: 197472099

Merge pull request tensorflow#771 from lukaszkaiser/push

1.6.2

Merge pull request tensorflow#751 from rsepassi/push

v1.6.1

Merge pull request tensorflow#736 from rsepassi/push

v1.6.0

Merge pull request tensorflow#708 from rsepassi/push

v1.5.7

Merge pull request tensorflow#686 from rsepassi/push

v1.5.6

Merge pull request tensorflow#646 from rsepassi/push

v1.5.5

Merge pull request tensorflow#633 from rsepassi/push

v1.5.4

Merge pull request tensorflow#620 from rsepassi/push

v1.5.3

Merge pull request tensorflow#593 from rsepassi/push

v1.5.2