
    BVh                        d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z
 ddlmZ  e
j                  dd	       ed
g        G d dej                                      Z e
j                  dd	       ed
g       G d dej                                       Z ej$                         reanead Z ej*                  e       y)z&Datasets for random number generators.    N)tf2)	v2_compat)dataset_ops)	random_op)deprecation)	tf_exportz"Use `tf.data.Dataset.random(...)`.zdata.experimental.RandomDataset)v1c                       e Zd ZdZy)RandomDatasetV2#A `Dataset` of pseudorandom values.N)__name__
__module____qualname____doc__     b/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/experimental/ops/random_ops.pyr   r      s     ,r   r   c                   b     e Zd ZdZ ej
                  ej                        d fd	       Z xZS )RandomDatasetV1r   c                 B    t        |      }t        t        |   |       y N)r   superr   __init__)selfseedwrapped	__class__s      r   r   zRandomDatasetV1.__init__&   s    d#G	/4)'2r   r   )	r   r   r   r   	functoolswrapsr   r   __classcell__)r   s   @r   r   r   !   s+     ,9???++,3 -3r   r   c                  F    t        j                         rt        ay t        ay r   )r   enabledr   RandomDatasetr   r   r   r   _tf2_callbackr$   2   s    [[]#M#Mr   )r   r   tensorflow.pythonr   tensorflow.python.compatr   tensorflow.python.data.opsr   r   tensorflow.python.utilr    tensorflow.python.util.tf_exportr   
deprecated_RandomDatasetr   DatasetV1Adapterr   r"   r#   r$   register_data_v2_callbackr   r   r   <module>r.      s    -  ! . 2 0 . 6 BC
,4,i.. , 5 D, BC
0123k22 3 3 D3 3;;=!-!-$ $	 # #M 2r   