B ٻd@@sXdZddlmZddlmZddlmZddlmZddlmZddlmZdd Z d S) z4Utilities for generating Tensor-valued random seeds.) constant_op)dtypes)ops) random_seed) array_ops)math_opsc Cst|\}}|dkr*tjdtjdd}ntj|tjdd}|dkrXtjdtjdd}n\tdL}tj|tjd}t j t t |dt |dtjdtjd||d}WdQRX||fS) a}Returns the local seeds an operation should use given an op-specific seed. See `random_seed.get_seed` for more details. This wrapper adds support for the case where `seed` may be a tensor. Args: seed: An integer or a `tf.int64` scalar tensor. Returns: A tuple of two `tf.int64` scalar tensors that should be used for the local seed of the calling dataset. Nrseed)dtypenameseed2)r i)r )rget_seedrZconstantrZint64rZconvert_to_tensorZ name_scoperZwhere_v2r logical_andequal)rr scoperY/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/data/util/random_seed.pyr s  r N) __doc__Ztensorflow.python.frameworkrrrrZtensorflow.python.opsrrr rrrrs