
    BVh=                     t    d 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 Z	 G d d	ej                        Zy
)z/The implementation of `tf.data.Dataset.repeat`.    )dataset_ops)constant_op)dtypes)ops)gen_dataset_opsc                     t        | ||      S N)_RepeatDataset)input_datasetcountnames      T/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/ops/repeat_op.py_repeatr      s    	ud	33    c                   $     e Zd ZdZd fd	Z xZS )r
   z1A `Dataset` that repeats its input several times.c                 `   || _         |,t        j                  dt        j                  d      | _        n+t        j                  |t        j                  d      | _        || _        t        j                  |j                  fd| j
                  i| j                  }t        | 9  ||       y)z#See `Dataset.repeat()` for details.Nr   )dtyper   )_input_datasetr   constantr   int64_countr   convert_to_tensor_namer   repeat_dataset_variant_tensor_common_argssuper__init__)selfr   r   r   variant_tensor	__class__s        r   r   z_RepeatDataset.__init__   s    'D}((6<<gNdk))
v||'3dkDJ$33%%kk 

N 
G]N3r   r	   )__name__
__module____qualname____doc__r   __classcell__)r"   s   @r   r
   r
      s    94 4r   r
   N)r&   tensorflow.python.data.opsr   tensorflow.python.frameworkr   r   r   tensorflow.python.opsr   r   UnaryUnchangedStructureDatasetr
    r   r   <module>r-      s0    6 2 3 . + 144[?? 4r   