o i@s,ddlZddlZddZddZddZdS)NcCtjj|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_nodetensorsr Y/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/torch/_lazy/computation.pyget_tensors_ts_device_data_nodesr cCr)z4Return the graph hash for the passed in lazy tensors)rr_lazy_get_graph_hashrr r r get_graph_hashsrcCstjj||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. )rrr_run_cached_graph)hash_str graph_inputsr r r run_cached_graphsr)torch._C._lazyrtorch._C._lazy_ts_backendr rrr r r r s