[tf.distribute] Adding a test for tf.data.Dataset.cache.
[tf.distribute] Adding a test for tf.data.Dataset.cache.

PiperOrigin-RevId: 307535019
Change-Id: I517f4ce2d4a12660cb827f0cbd0e0e938d71b7e9

gunan/tensorflow,gunan/tensorflow,gunan/tensorflow,gunan/tensorflow,gunan/tensorflow,gunan/tensorflow,gunan/tensorflow,gunan/tensorflow,gunan/tensorflow,gunan/tensorflow,gunan/tensorflow,gunan/tensorflow