o
    bi                     @   s0   d dl Z d dlmZ dd Zdd Zdd ZdS )	    N)global_statec                 C   s2   t | d}|d ur| |v r|| = d S d S d S N_dict)r   get_global_attribute)	tensor_idattr	attr_dict r	   ^/home/ubuntu/.local/lib/python3.10/site-packages/keras/src/backend/common/tensor_attributes.py_clear_tensor_attr   s   
r   c              	   C   s   z	t | || W d S  tyV   t| d}|d u r.|d u r#Y d S i }t| d| |d urE||t| < t| tt| | Y d S t| |v rS|t| = Y d S Y d S w r   )	setattrAttributeErrorr   r   set_global_attributeidweakreffinalizer   )tensorr   valuer   r	   r	   r
   set_tensor_attr   s    r   c                 C   sB   t | |st| d}|d ur|t| d S d S t| |d S r   )hasattrr   r   getr   getattr)r   r   r   r	   r	   r
   get_tensor_attr   s   
r   )r   keras.src.backend.commonr   r   r   r   r	   r	   r	   r
   <module>   s
    