Adds gshard params for SimpleFullSoftmax and SimpleEmbeddingLayer.
Adds gshard params for SimpleFullSoftmax and SimpleEmbeddingLayer.

PiperOrigin-RevId: 350458589

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