Searched refs:use_side_stream_for_tensor_copies (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | utils.py | 99 inputs: S, target_device: torch.device, use_side_stream_for_tensor_copies: bool
106 inputs: T, target_device: torch.device, use_side_stream_for_tensor_copies: bool
111 def _recursive_to(inputs, target_device, use_side_stream_for_tensor_copies): argument
119 if not use_side_stream_for_tensor_copies:
264 use_side_stream_for_tensor_copies: bool,
267 _recursive_to(inputs, target_device, use_side_stream_for_tensor_copies)
272 _recursive_to(kwargs, target_device, use_side_stream_for_tensor_copies)
|
| /aosp_15_r20/external/pytorch/torch/nn/parallel/ |
| H A D | distributed.py | 805 self.use_side_stream_for_tensor_copies = (
1546 self.use_side_stream_for_tensor_copies,
1656 self.use_side_stream_for_tensor_copies,
|