Add generic_input op unit test for usage with tf.data
Add generic_input op unit test for usage with tf.data

This reproduces the issue encountered when trying to create a tf.data.Dataset that
invokes generic_input.

PiperOrigin-RevId: 324725586

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