Refactor CreateTpuEmbeddingEnqueueOps, and support the case where use_per_host_infeed=False.
Refactor CreateTpuEmbeddingEnqueueOps, and support the case where
use_per_host_infeed=False.

PiperOrigin-RevId: 383691903

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