Check that the TpuEmbeddingLayer configuration matches existing TF TPUEmbedding object before reusing. Also fix a bug where the load/retrieve ops are added multiple times.
Check that the TpuEmbeddingLayer configuration matches existing TF TPUEmbedding object before reusing.
Also fix a bug where the load/retrieve ops are added multiple times.

PiperOrigin-RevId: 368669145

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