The loss weights should have same dtype as the loss (typically tf.float32) in _ComputeSoftmaxMetrics().
The loss weights should have same dtype as the loss (typically tf.float32) in
_ComputeSoftmaxMetrics().

PiperOrigin-RevId: 354500462

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