Add additional encoder inference graph support for beam search decoders.
Add additional encoder inference graph support for beam search decoders.

PiperOrigin-RevId: 324734810

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