TPU Embedding load/retrieve parameters ops need to be declared in the outer graph scope with control flow v2.
TPU Embedding load/retrieve parameters ops need to be declared in the
outer graph scope with control flow v2.

PiperOrigin-RevId: 320328260

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