
    ɯwg                     &    d dl Zd dlZd Zd Zd Zy)    Nc                 T    t         j                  j                  j                  |       S )zReturn 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tensorss    \/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/torch/_lazy/computation.pyget_tensors_ts_device_data_noder      s     88$$EEgNN    c                 T    t         j                  j                  j                  |       S )z4Return the graph hash for the passed in lazy tensors)r   r   _lazy_get_graph_hashr   s    r
   get_graph_hashr      s    88>>))'22r   c                 V    t         j                  j                  j                  | |      S )zRunning 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_inputss     r
   run_cached_graphr      s!     88$$66xNNr   )torch._C._lazyr   torch._C._lazy_ts_backendr   r   r    r   r
   <module>r      s      O3
Or   