o
    i eS                     @   sj   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l
mZ dd	 ZG d
d dejZdS )z.The implementation of `tf.data.Dataset.cache`.    )tf2)dataset_ops)context)dtypes)ops)gen_dataset_opsc                 C   s   t | ||S N)CacheDataset)input_datasetfilenamename r   R/var/www/myenv/lib/python3.10/site-packages/tensorflow/python/data/ops/cache_op.py_cache   s   r   c                       s"   e Zd ZdZd fdd	Z  ZS )r	   z.A `Dataset` that caches elements of its input.Nc                    s   || _ tj|tjdd| _|| _t r.t	
 st r.tj|jf| jt d| j}ntj|jfd| ji| j}t || dS )z"See `Dataset.cache()` for details.r   )dtyper   )r   cacheN)_input_datasetr   convert_to_tensorr   string	_filename_namer   enabledr   executing_eagerlyinside_functionr   cache_dataset_v2_variant_tensordummy_memory_cache_common_argscache_datasetsuper__init__)selfr
   r   r   variant_tensor	__class__r   r   r        s,   zCacheDataset.__init__r   )__name__
__module____qualname____doc__r    __classcell__r   r   r#   r   r	      s    r	   N)r(   tensorflow.pythonr   tensorflow.python.data.opsr   tensorflow.python.eagerr   tensorflow.python.frameworkr   r   tensorflow.python.opsr   r   UnaryUnchangedStructureDatasetr	   r   r   r   r   <module>   s   