As support for int64 global steps is not ready in TPUs, I am reversing this change so that our internal performance and regression tests will pass.