Remove tf.identity, explicitly place embeddings on CPU for eval
Remove tf.identity, explicitly place embeddings on CPU for eval

PiperOrigin-RevId: 383504784

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