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,
|