o
    ûÇh—  ã                   @   s,   d dl Zd dlZdd„ Zdd„ Zdd„ ZdS )é    Nc                 C   ó   t jj | ¡S )zæReturn tensor ids and eager tensors for DeviceData nodes in the
    IR for the passed in lazy tensors.

    TODO: This API is currently ts backend specific. We are working on
    generalizing it to all backends including XLA.
    )ÚtorchÚ_CÚ_lazy_ts_backendÚ _get_tensors_ts_device_data_node©Útensors© r	   úk/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/_lazy/computation.pyÚget_tensors_ts_device_data_node   s   r   c                 C   r   )z4Return the graph hash for the passed in lazy tensors)r   r   Ú_lazyÚ_get_graph_hashr   r	   r	   r
   Úget_graph_hash   s   r   c                 C   s   t jj | |¡S )zºRunning the cached computation graph with the given inputs

    TODO: This API is currently ts backend specific. We are working on
    generalizing it to all backends including XLA.
    )r   r   r   Ú_run_cached_graph)Úhash_strÚgraph_inputsr	   r	   r
   Úrun_cached_graph   s   r   )Útorch._C._lazyr   Útorch._C._lazy_ts_backendr   r   r   r	   r	   r	   r
   Ú<module>   s
   
