Log expert_gating once it is been masked by the importance tensor to be sure no padded probabilities are being logged.
Log expert_gating once it is been masked by the importance tensor to be sure no padded probabilities are being logged.

PiperOrigin-RevId: 378321668

tensorflow/mesh,tensorflow/mesh