Fix masking generate logic in sequence model.
sequence model was incorrectly using the layer output instead of input to generate mask, and compute_mask of the layer usually does not have shape check. PiperOrigin-RevId: 227605298
Loading
Please sign in to comment