Increase shard count on extract_image_patches_grad_test.

The test sometimes would time out on the nightly MacOS builds.

PiperOrigin-RevId: 353279090
Change-Id: I33e29f73ef0285ff6a674015538707de1d7e5144
This commit is contained in:
Reed Wanderman-Milne 2021-01-22 11:11:48 -08:00 committed by TensorFlower Gardener
parent c3767f71b8
commit 3516dc97f3

View File

@ -3231,7 +3231,7 @@ cuda_py_test(
name = "extract_image_patches_grad_test",
size = "medium",
srcs = ["extract_image_patches_grad_test.py"],
shard_count = 9,
shard_count = 15,
tags = ["notap"], # http://b/31080670
deps = [
"//tensorflow/python:array_ops",