Change the order of iterative process and evaluation function to avoid interleaving graph and eager mode.
Change the order of iterative process and evaluation function to avoid interleaving graph and eager mode.

PiperOrigin-RevId: 379533029
Change-Id: Ifb3cb8c84d24866e9745563d95abc7a3eb007c5c

google-research/federated,google-research/federated