Increase cholesky_op_test to medium, bump shard_count 1 more.

PiperOrigin-RevId: 157635774
This commit is contained in:
A. Unique TensorFlower 2017-05-31 14:31:24 -07:00 committed by TensorFlower Gardener
parent bef563dc88
commit 88ffe62767

View File

@ -122,7 +122,7 @@ tf_py_test(
cuda_py_test(
name = "cholesky_op_test",
size = "small",
size = "medium",
srcs = ["cholesky_op_test.py"],
additional_deps = [
"//third_party/py/numpy",
@ -133,7 +133,7 @@ cuda_py_test(
"//tensorflow/python:math_ops",
"//tensorflow/python:platform",
],
shard_count = 4,
shard_count = 5,
)
tf_py_test(