Make positional embedding layers use inferred shape values for time tensor.
Make positional embedding layers use inferred shape values for time tensor.

PiperOrigin-RevId: 219873903

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