Fix flaky test gpu_xla_dct_ops_test by increasing the number of shards

PiperOrigin-RevId: 350289211 Change-Id: Ibed1568c0dcf312f7fbfac5207f036900c5bf25e
2021-01-05 23:00:28 -08:00 · 2021-01-05 23:00:28 -08:00 · 934d476896
commit 934d476896
parent 78012e5d6c
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/python/kernel_tests/signal/BUILD
+++ b/tensorflow/python/kernel_tests/signal/BUILD
@ -25,6 +25,7 @@ cuda_py_tests(
    name = "dct_ops_test",
    srcs = ["dct_ops_test.py"],
    python_version = "PY3",
+    shard_count = 16,
    deps = [
        "//tensorflow/python:client_testlib",
        "//tensorflow/python:framework_for_generated_wrappers",