| commit | 50ce5d05c382c32da7289d506c13da2b67912fdf | [log] [tgz] |
|---|---|---|
| author | Yang Chen <yangchen@google.com> | Tue May 25 11:10:25 2021 -0700 |
| committer | TensorFlower Gardener <gardener@tensorflow.org> | Tue May 25 11:20:16 2021 -0700 |
| tree | b2dded2cbc5bb1ac64959b10da418a65c80b7428 | |
| parent | 0d8a75e0cb2b3a504869e75c37a8b1c218501887 [diff] |
[tf.data] Fix flaky test data/kernel_tests:memory_cleanup_test. PiperOrigin-RevId: 375748975 Change-Id: Ic51aa3b13fe95401c7138c4ffefbfdad5777f0eb
diff --git a/tensorflow/python/data/kernel_tests/BUILD b/tensorflow/python/data/kernel_tests/BUILD index c6803f4..b419ced 100644 --- a/tensorflow/python/data/kernel_tests/BUILD +++ b/tensorflow/python/data/kernel_tests/BUILD
@@ -569,7 +569,7 @@ name = "memory_cleanup_test", size = "small", srcs = ["memory_cleanup_test.py"], - shard_count = 2, + shard_count = 12, tags = [ "no_cuda_asan", # TODO(b/183963686) "notsan", # TODO(b/144706539)