Home
last modified time | relevance | path

Searched refs:_thread_local_tensor_tables (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/rpc/
H A Dinternal.py20 _thread_local_tensor_tables = threading.local() variable
57 global _thread_local_tensor_tables
58 return _thread_local_tensor_tables.recv_tables[tensor_index]
61 global _thread_local_tensor_tables
62 _thread_local_tensor_tables.send_tables.append(tensor)
63 tensor_index = len(_thread_local_tensor_tables.send_tables) - 1
129 global _thread_local_tensor_tables
130 if hasattr(_thread_local_tensor_tables, "send_tables"):
131 old_send_tables = _thread_local_tensor_tables.send_tables
134 _thread_local_tensor_tables.send_tables = []
[all …]