diff options
| author |  Peter Hawkins <phawkins@google.com> | 2017-10-13 06:55:45 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-13 06:59:56 -0700 |
| commit | a3b2d6f395ef3f66c9ccd8578e94243e49f76576 (patch) | |
| tree | a89ab2e4435f8455882e4f230c7a4897ac7cec2a | |
| parent | 7e31a198c8a6a6a618ca959a69941bcdd82cb140 (diff) | |
[TF:XLA] Add bound to ArgMax in randomized tests.
PiperOrigin-RevId: 172091245
| -rw-r--r-- | tensorflow/compiler/tests/randomized_tests.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/tests/randomized_tests.cc b/tensorflow/compiler/tests/randomized_tests.cc index 56e10a1587..5129171cd4 100644 --- a/tensorflow/compiler/tests/randomized_tests.cc +++ b/tensorflow/compiler/tests/randomized_tests.cc @@ -899,7 +899,7 @@ TEST_F(OpTest, ApproximateEqual) { TEST_F(OpTest, ArgMax) { Repeatedly([this]() { - std::vector<int64> dims = RandomDims(1, 5); + std::vector<int64> dims = RandomDims(1, 5, 1); int num_dims = dims.size(); int reduce_dim = std::uniform_int_distribution<int32>(-num_dims, num_dims)(generator()); |