diff options
Diffstat (limited to 'tensorflow/python/data/util/random_seed.py')
| -rw-r--r-- | tensorflow/python/data/util/random_seed.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/python/data/util/random_seed.py b/tensorflow/python/data/util/random_seed.py index d5169f7a53..d24df6d957 100644 --- a/tensorflow/python/data/util/random_seed.py +++ b/tensorflow/python/data/util/random_seed.py @@ -24,6 +24,7 @@ from tensorflow.python.framework import ops from tensorflow.python.framework import random_seed from tensorflow.python.ops import array_ops from tensorflow.python.ops import math_ops +from tensorflow.python.ops import random_ops def get_seed(seed): @@ -37,7 +38,7 @@ def get_seed(seed): Returns: A tuple of two `tf.int64` scalar tensors that should be used for the local - seed of the calling dataset. + seeds of the calling dataset. """ seed, seed2 = random_seed.get_seed(seed) if seed is None: @@ -45,7 +46,7 @@ def get_seed(seed): else: seed = ops.convert_to_tensor(seed, dtype=dtypes.int64, name="seed") if seed2 is None: - seed2 = constant_op.constant(0, dtype=dtypes.int64, name="seed2") + seed2 = random_ops.random_uniform([], 1, 2**63 - 1, dtype=dtypes.int64) else: with ops.name_scope("seed2") as scope: seed2 = ops.convert_to_tensor(seed2, dtype=dtypes.int64) |