Allow data and project matrix with different dtypes
Allow data and project matrix with different dtypes

PiperOrigin-RevId: 351257931

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