o
    2hK                     @   sf   d Z ddlZddlZddlmZ G dd dejZe Zedg dejdd	 Z	d
d Z
dd ZdS )z5Context for storing options for loading a SavedModel.    N)	tf_exportc                       s@   e Zd ZdZ fddZdd Zdd Zdd	 Zd
d Z  Z	S )LoadContextzA context for loading a model.c                    s   t    g | _d | _d S N)super__init___entered_load_context_load_optionsself	__class__ d/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/distribute/load_context.pyr      s   

zLoadContext.__init__c                 C   s   || _ | jd d S )NT)r   r   append)r
   load_optionsr   r   r   set_load_options   s   zLoadContext.set_load_optionsc                 C   s   d | _ | j  d S r   )r   r   popr	   r   r   r   clear_load_options#   s   zLoadContext.clear_load_optionsc                 C      | j S r   )r   r	   r   r   r   r   '      zLoadContext.load_optionsc                 C   r   r   )r   r	   r   r   r   in_load_context*   r   zLoadContext.in_load_context)
__name__
__module____qualname____doc__r   r   r   r   r   __classcell__r   r   r   r   r      s    r   z__internal__.load_context)v1c                 c   s,    t |  z
d V  W t   d S t   w r   )_load_contextr   r   )r   r   r   r   load_context1   s
   
r   c                   C      t  S )z.Returns the load options under a load context.)r   r   r   r   r   r   get_load_options;      r    c                   C   r   )z%Returns whether under a load context.)r   r   r   r   r   r   r   @   r!   r   )r   
contextlib	threading tensorflow.python.util.tf_exportr   localr   r   contextmanagerr   r    r   r   r   r   r   <module>   s   
