Fix assert message in beam search op Assert compares .dim_size() (dimension size) but prints .dims() (rank)
Fix assert message in beam search op
Assert compares .dim_size() (dimension size) but prints .dims() (rank)

PiperOrigin-RevId: 252482834

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