Ensure that projection layer has an int32 shape tensor.
Ensure that projection layer has an int32 shape tensor.

PiperOrigin-RevId: 378681563

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