tree b2dded2cbc5bb1ac64959b10da418a65c80b7428
parent 0d8a75e0cb2b3a504869e75c37a8b1c218501887
author Yang Chen <yangchen@google.com> 1621966225 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1621966816 -0700

[tf.data] Fix flaky test data/kernel_tests:memory_cleanup_test.

PiperOrigin-RevId: 375748975
Change-Id: Ic51aa3b13fe95401c7138c4ffefbfdad5777f0eb