Fix to support DecodeStep in MTDecoderV1 when packed_input=True
Fix to support DecodeStep in MTDecoderV1 when packed_input=True

PiperOrigin-RevId: 210028916

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