aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/linear_optimizer
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-04-25 19:18:53 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-25 19:24:46 -0700
commit5a751a41aa92068ca0dbf991524cb4bed90c630f (patch)
tree462dab8a4085f942c64ecaeb6412892fd96a9958 /tensorflow/contrib/linear_optimizer
parent752984871bb8a89e774f1c329d42b3bb7b7d7b3f (diff)
Reduce number of combinations that are tested
PiperOrigin-RevId: 194335483
Diffstat (limited to 'tensorflow/contrib/linear_optimizer')
-rw-r--r--tensorflow/contrib/linear_optimizer/python/kernel_tests/sdca_ops_test.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/linear_optimizer/python/kernel_tests/sdca_ops_test.py b/tensorflow/contrib/linear_optimizer/python/kernel_tests/sdca_ops_test.py
index 6e6c812adc..b5741967ab 100644
--- a/tensorflow/contrib/linear_optimizer/python/kernel_tests/sdca_ops_test.py
+++ b/tensorflow/contrib/linear_optimizer/python/kernel_tests/sdca_ops_test.py
@@ -39,8 +39,8 @@ from tensorflow.python.ops import variables as variables_lib
from tensorflow.python.platform import googletest
_MAX_ITERATIONS = 100
-_SHARD_NUMBERS = [None, 1, 3, 10]
-_NUM_LOSS_PARTITIONS = [2, 4]
+_SHARD_NUMBERS = [None, 1, 3]
+_NUM_LOSS_PARTITIONS = [4]
def make_example_proto(feature_dict, target, value=1.0):