
    ij                     h    d Z ddlZddlmZ  G d dej                        Z e       Z G d d      Zy)z,Helper functions for TPU Embedding Contexts.    N)loggingc                        e Zd Z fdZ xZS )_EmbeddingPipeliningStatec                 0    t         |           d| _        y )NT)super__init__enabled)self	__class__s    /var/www/html/emotional.easysim.app/public_html/venv/lib/python3.12/site-packages/tensorflow/python/tpu/embedding_context_utils.pyr   z"_EmbeddingPipeliningState.__init__   s    	GDL    )__name__
__module____qualname__r   __classcell__)r   s   @r   r   r      s     r   r   c                   "    e Zd ZdZd Zd Zd Zy)SequentialEmbeddingContextz?Disables embedding pipelining for all ops created in the scope.c                 .    t         j                  | _        y )N)embedding_pipelining_stater	   ,_original_embedding_pipelining_state_enabledr
   s    r   r   z#SequentialEmbeddingContext.__init__!   s    "** 	5r   c                 D    dt         _        t        j                  d       y )NFz$Entering SequentialEmbeddingContext.)r   r	   r   infor   s    r   	__enter__z$SequentialEmbeddingContext.__enter__&   s    ).&LL78r   c                 X    | j                   t        _        t        j                  d       y )Nz#Exiting SequentialEmbeddingContext.)r   r   r	   r   r   )r
   exc_typeexc_valexc_tbs       r   __exit__z#SequentialEmbeddingContext.__exit__*   s"    99 & LL67r   N)r   r   r   __doc__r   r   r    r   r   r   r      s    G
98r   r   )r    	threadingabslr   localr   r   r   r!   r   r   <module>r%      s5    3  	  78 8 8r   