
    BVhK                         d Z ddlZddlZddlmZ  G d dej
                        Z e       Z edg       ej                  d               Z	d	 Z
d
 Zy)z5Context for storing options for loading a SavedModel.    N)	tf_exportc                   :     e Zd ZdZ fdZd Zd Zd Zd Z xZ	S )LoadContextzA context for loading a model.c                 >    t         |           g | _        d | _        y N)super__init___entered_load_context_load_options)self	__class__s    Y/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/distribute/load_context.pyr	   zLoadContext.__init__   s    	G!#DD    c                 H    || _         | j                  j                  d       y )NT)r   r
   append)r   load_optionss     r   set_load_optionszLoadContext.set_load_options   s    %D%%d+r   c                 F    d | _         | j                  j                          y r   )r   r
   popr   s    r   clear_load_optionszLoadContext.clear_load_options#   s    D""$r   c                     | j                   S r   )r   r   s    r   r   zLoadContext.load_options'   s    r   c                     | j                   S r   )r
   r   s    r   in_load_contextzLoadContext.in_load_context*   s    %%%r   )
__name__
__module____qualname____doc__r	   r   r   r   r   __classcell__)r   s   @r   r   r      s    &
,%&r   r   z__internal__.load_context)v1c              #      K   t         j                  |        	 d  t         j                          y # t         j                          w xY wwr   )_load_contextr   r   )r   s    r   load_contextr#   1   s6        .'	$$&M$$&s   A2 AAAc                  *    t         j                         S )z.Returns the load options under a load context.)r"   r    r   r   get_load_optionsr&   ;   s    		#	#	%%r   c                  *    t         j                         S )z%Returns whether under a load context.)r"   r   r%   r   r   r   r   @   s    		&	&	((r   )r   
contextlib	threading tensorflow.python.util.tf_exportr   localr   r"   contextmanagerr#   r&   r   r%   r   r   <module>r-      s`    <   6&)// &.  &2.'  /'&
)r   