Makes the assertion in Dtypes work for TPUMirroredVariable (and other objects that support 'dtype').
Makes the assertion in Dtypes work for TPUMirroredVariable (and other objects that support 'dtype').

PiperOrigin-RevId: 272953359

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