
    BVhI                         d Z ddlZddlZ G d dej                        Z e       Zej                  d        Zd Zd Z	y)z Context for building SavedModel.    Nc                   :     e Zd ZdZ fdZd Zd Zd Zd Z xZ	S )SaveContextz-A context for building a graph of SavedModel.c                 F    t         t        |           d| _        d | _        y NF)superr   __init___in_save_context_options)self	__class__s    Z/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/saved_model/save_context.pyr   zSaveContext.__init__   s    	+t%'!DDM    c                 P    | j                         st        d      | j                  S )NzNot in a SaveContext.)in_save_context
ValueErrorr
   r   s    r   optionszSaveContext.options   s$    !.//==r   c                      d| _         || _        y )NTr	   r
   )r   r   s     r   enter_save_contextzSaveContext.enter_save_context"   s     DDMr   c                      d| _         d | _        y r   r   r   s    r   exit_save_contextzSaveContext.exit_save_context&   s    !DDMr   c                     | j                   S )N)r	   r   s    r   r   zSaveContext.in_save_context*   s       r   )
__name__
__module____qualname____doc__r   r   r   r   r   __classcell__)r   s   @r   r   r      s    5

!r   r   c              #      K   t               rt        d      t        j                  |        	 d  t        j	                          y # t        j	                          w xY ww)NzAlready in a SaveContext.)r   r   _save_contextr   r   )r   s    r   save_contextr!   0   sF     
0
11""7+&	##%M##%s   +A A A AA c                  *    t         j                         S )z%Returns whether under a save context.)r    r    r   r   r   r   ;   s    		&	&	((r   c                  *    t         j                         S )z1Returns the save options if under a save context.)r    r   r#   r   r   get_save_optionsr%   @   s    				  r   )
r   
contextlib	threadinglocalr   r    contextmanagerr!   r   r%   r#   r   r   <module>r*      sM    '  !)// !0  & &)
!r   