Migrate quarantine.defun to tf.function
Migrate quarantine.defun to tf.function

PiperOrigin-RevId: 484306368

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