Let GpipeEmbedding layer pass task_ids to the follo-wup layers.
Let GpipeEmbedding layer pass task_ids to the follo-wup layers.

PiperOrigin-RevId: 289028570

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