diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-09-06 14:02:42 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-06 14:10:51 -0700 |
commit | 6c7d1fe3d64b0fc0f7bfd44b9b3cdaf3939b1ff0 (patch) | |
tree | 45de9306beb713b1c92783efbbe495895e4e4f4f /tensorflow/contrib/resampler | |
parent | 570147d34015cb8c36869a548b7eac0409416601 (diff) |
Reduce batch size for resampler backward pass test, to speed up test.
PiperOrigin-RevId: 167769539
Diffstat (limited to 'tensorflow/contrib/resampler')
-rw-r--r-- | tensorflow/contrib/resampler/python/ops/resampler_ops_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/resampler/python/ops/resampler_ops_test.py b/tensorflow/contrib/resampler/python/ops/resampler_ops_test.py index 9aa1e05628..6253f96315 100644 --- a/tensorflow/contrib/resampler/python/ops/resampler_ops_test.py +++ b/tensorflow/contrib/resampler/python/ops/resampler_ops_test.py @@ -163,7 +163,7 @@ class ResamplerTest(test.TestCase): data_channels = 3 warp_width = 2 warp_height = 6 - batch_size = 10 + batch_size = 3 warp = _make_warp(batch_size, warp_height, warp_width, dtype.as_numpy_dtype) data_shape = (batch_size, data_height, data_width, data_channels) |