Make contrib.seq2seq more compatible with tf 2.0. (1/6)

1. Add a new loss object which wrap around the existing loss function, which
use can use it keras model.
2. Slightly update the code for readability and reduce duplication.
3. Update the test case to be eager compatible.

PiperOrigin-RevId: 227860442