Add MeshSplit to SharedEmbeddingLayer embedding outputs following SimpleEmbeddingLayer.
Add MeshSplit to SharedEmbeddingLayer embedding outputs following
SimpleEmbeddingLayer.

PiperOrigin-RevId: 375626590

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