diff options
| author |  Peter Hawkins <phawkins@google.com> | 2018-10-02 07:34:40 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-02 07:38:53 -0700 |
| commit | 97d515273a1e86a861cdfb338671a42b3b1126a7 (patch) | |
| tree | b8eb3750be7468eda2444dafd8bc7192b84f1ee6 /tensorflow/compiler | |
| parent | 35f3046a326daea0179d024044636f2fcbb45f4a (diff) | |
Make StatelessRandomOpsTest.testRandomNormalIsFinite actually test stateless_random_normal.
Fixes #22611
PiperOrigin-RevId: 215385610
Diffstat (limited to 'tensorflow/compiler')
| -rw-r--r-- | tensorflow/compiler/tests/stateless_random_ops_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/tests/stateless_random_ops_test.py b/tensorflow/compiler/tests/stateless_random_ops_test.py index f3861043b2..e8741bc468 100644 --- a/tensorflow/compiler/tests/stateless_random_ops_test.py +++ b/tensorflow/compiler/tests/stateless_random_ops_test.py @@ -91,7 +91,7 @@ class StatelessRandomOpsTest(xla_test.XLATestCase): with self.cached_session() as sess, self.test_scope(): for dtype in self._random_types(): seed_t = array_ops.placeholder(dtypes.int32, shape=[2]) - x = stateless.stateless_random_uniform( + x = stateless.stateless_random_normal( shape=[10000], seed=seed_t, dtype=dtype) y = sess.run(x, {seed_t: [0x12345678, 0xabcdef12]}) self.assertTrue(np.all(np.isfinite(y))) |