tree 0531d5fa8f796b228423842201b9a034aeac75cb
parent e4a80d5e7cb593dbd24fa23bdffecc04fc8e83ea
author Jay Shi <xiaojies@google.com> 1601929647 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1601930413 -0700

[tf.data] Only apply the optimization `enable_gradient_descent` on the main dataset. Also update the unit test.

PiperOrigin-RevId: 335486882
Change-Id: I864f44d90fc3d84f702b3e53ed6b226362dd27a7