Replace tf.trainable_variables() with a list comprehension since it doesn't work in eager mode.
Replace tf.trainable_variables() with a list comprehension since it doesn't work in eager mode.

PiperOrigin-RevId: 383947862

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