o
    i e                     @   s*   d Z ddlmZ ddlmZ dddZdS )z0The implementation of `tf.data.Dataset.counter`.    )dataset_ops)opsNc                    st   t d+ t j| |dd} t j |dd tjjd|dd |  fddW  d    S 1 s3w   Y  d S )	Ncounterstart)dtypenamestepr   )r   c                    s   |   | fS N )state_r   r
   T/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/data/ops/counter_op.py<lambda>   s    z_counter.<locals>.<lambda>)r   
name_scopeconvert_to_tensorr   Datasetfrom_tensorsrepeatscan)r   r   r   r   r
   r   r   _counter   s   $r   r	   )__doc__tensorflow.python.data.opsr   tensorflow.python.frameworkr   r   r
   r
   r
   r   <module>   s   