Reset the step seed at the beginning of each tf.function to mimic Defun's behavior, so that the original step seed from parent graph won't get used as an unexpected implicit capture.
Reset the step seed at the beginning of each tf.function to mimic Defun's
behavior, so that the original step seed from parent graph won't get used as an
unexpected implicit capture.

PiperOrigin-RevId: 326522496

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo