Updating to use newer TF version.
Updating to use newer TF version.

PiperOrigin-RevId: 241998896

google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research