diff options
author | Yao Zhang <yaozhang@google.com> | 2018-02-07 18:11:51 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-07 18:15:52 -0800 |
commit | 78e4ed153a853536622ff606fc5f6c48a1573ac6 (patch) | |
tree | 4faaa7d87676a98ced03bb26c0548c55499d125c | |
parent | ddf9536dfeb358237219e27c185c729fd4c8537b (diff) |
Reduce the number of concats to avoid test timeout.
PiperOrigin-RevId: 184929151
-rw-r--r-- | tensorflow/python/kernel_tests/concat_op_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/concat_op_test.py b/tensorflow/python/kernel_tests/concat_op_test.py index a5fd3bc334..127bc6bb20 100644 --- a/tensorflow/python/kernel_tests/concat_op_test.py +++ b/tensorflow/python/kernel_tests/concat_op_test.py @@ -495,9 +495,9 @@ class ConcatOpTest(test.TestCase): p = [] shape = np.array([7, 13]) if test.is_gpu_available(): - num_tensors = 10000 + num_tensors = 5000 else: - num_tensors = 1000 + num_tensors = 500 for i in np.arange(num_tensors): input_shape = shape placeholder = array_ops.placeholder(dtypes.float32, shape=input_shape) |