make multi-source step by step decoder compatible with constrained_alignment
make multi-source step by step decoder compatible with constrained_alignment

PiperOrigin-RevId: 299150786

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