BVh@VdZddlmZddlmZddlmZddlmZddlmZddlmZdZ y ) z4Utilities for generating Tensor-valued random seeds.) constant_op)dtypes)ops) random_seed) array_ops)math_opsc tj|\}}|'tjdtj d}n&t j|tj d}|*tjdtj d}||fSt jd5}t j|tj }tjtjtj|dtj|dtjdtj ||}ddd||fS#1swY||fSxYw) 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_seedrconstantrint64rconvert_to_tensor name_scoperwhere_v2r logical_andequal)r r scopes W/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/util/random_seed.pyrrs$$T*+$ \   F CD  V\\ GD ]  &,,W EE u  E##E>e    nnT1%x~~eQ'? A   y =  e u us )BEEN) __doc__tensorflow.python.frameworkrrrrtensorflow.python.opsrrrrrs!;3.+3+*r