o
    ?e=                     @   s^   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d Z	G d	d
 d
ej
ZdS )z/The implementation of `tf.data.Dataset.repeat`.    )dataset_ops)constant_op)dtypes)ops)gen_dataset_opsc                 C   s   t | ||S N)_RepeatDataset)input_datasetcountname r   e/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/data/ops/repeat_op.py_repeat   s   r   c                       s"   e Zd ZdZd fdd	Z  ZS )r   z1A `Dataset` that repeats its input several times.Nc                    sl   || _ |du rtjdtjdd| _n
tj|tjdd| _|| _t	j
|jfd| ji| j}t || dS )z#See `Dataset.repeat()` for details.Nr
   )Zdtyper   )Z_input_datasetr   Zconstantr   Zint64_countr   Zconvert_to_tensor_namer   Zrepeat_datasetZ_variant_tensorZ_common_argssuper__init__)selfr	   r
   r   Zvariant_tensor	__class__r   r   r      s   z_RepeatDataset.__init__r   )__name__
__module____qualname____doc__r   __classcell__r   r   r   r   r      s    r   N)r   Ztensorflow.python.data.opsr   Ztensorflow.python.frameworkr   r   r   Ztensorflow.python.opsr   r   ZUnaryUnchangedStructureDatasetr   r   r   r   r   <module>   s   