Home
last modified time | relevance | path

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 Dutils.py99 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 Ddistributed.py805 self.use_side_stream_for_tensor_copies = (
1546 self.use_side_stream_for_tensor_copies,
1656 self.use_side_stream_for_tensor_copies,