a ==icˆã@s,ddlZddlZdd„Zdd„Zdd„ZdS)éNcCstjj |¡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_backendZ _get_tensors_ts_device_data_node©Ztensors©rúh/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/torch/_lazy/computation.pyÚget_tensors_ts_device_data_nodesrcCstjj |¡S)z4Return the graph hash for the passed in lazy tensors)rrZ_lazyZ_get_graph_hashrrrrÚget_graph_hash sr cCstjj ||¡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. )rrrZ_run_cached_graph)Zhash_strZ graph_inputsrrrÚrun_cached_graphsr )Ztorch._C._lazyrZtorch._C._lazy_ts_backendrr r rrrrÚs