o
    ?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   d/home/www/facesmatcher.com/pyenv/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   )Zdtyper   )r   cacheN)Z_input_datasetr   Zconvert_to_tensorr   string	_filename_namer   enabledr   Zexecuting_eagerlyZinside_functionr   Zcache_dataset_v2Z_variant_tensorZdummy_memory_cacheZ_common_argsZcache_datasetsuper__init__)selfr
   r   r   Z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   Ztensorflow.pythonr   Ztensorflow.python.data.opsr   Ztensorflow.python.eagerr   Ztensorflow.python.frameworkr   r   Ztensorflow.python.opsr   r   ZUnaryUnchangedStructureDatasetr	   r   r   r   r   <module>   s   