Rollback of changes to mtf.top_1 .  There seem to be some bad results both ways. TODO(noam): fix this.
Rollback of changes to mtf.top_1 .  There seem to be some bad results
both ways.
TODO(noam): fix this.

PiperOrigin-RevId: 270824423

tensorflow/mesh,tensorflow/mesh