Increase shard_count for extract_image_patches_grad_test.

The test was previously timing out. PiperOrigin-RevId: 255441082
huangjin1995 · Jun 27, 2019 · b764c5d · b764c5d
1 parent 298f7e9
commit b764c5d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD
@@ -3244,6 +3244,7 @@ cuda_py_test(
         "//tensorflow/python:framework",
         "//tensorflow/python:framework_for_generated_wrappers",
     ],
+    shard_count = 3,
     tags = ["notap"],  # http://b/31080670
     xla_enable_strict_auto_jit = True,
 )