Turn off autograph on remaining defun in model_tuils.py until TFF moves to a newer TensorFlow version.
Turn off autograph on remaining defun in model_tuils.py until TFF moves to a newer TensorFlow version.

PiperOrigin-RevId: 231479136

tensorflow/federated,tensorflow/federated,tensorflow/federated