| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/ |
| H A D | __init__.py | 34 process_group=None, argument
89 process_group=None, argument
143 process_group=None, argument
199 process_group=None, argument
253 process_group=None, argument
309 process_group=None, argument
359 local_shards: List[Shard], *global_size, process_group=None, init_rrefs=False argument
|
| H A D | api.py | 253 process_group=None, argument
287 def _prepare_init(self, process_group=None, init_rrefs=False): argument
495 self, memory_format=torch.preserve_format, process_group=None argument
554 process_group=None, argument
720 process_group=None, argument
881 process_group=None, argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/ |
| H A D | api.py | 17 tensor: torch.Tensor, sharding_spec: ShardingSpec, src_rank=0, process_group=None argument
82 process_group=None, argument
136 def load_with_process_group(process_group): argument
226 def shard_module(module: nn.Module, plan: ShardingPlan, src_rank=0, process_group=None): argument
|
| H A D | op_registry_utils.py | 36 def wrapper(types, args, kwargs, process_group): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/ |
| H A D | binary_cmp.py | 24 def binary_cmp(cmp_fun, types, args, kwargs=None, process_group=None): argument
73 def equal(types, args, kwargs, process_group): argument
78 def allclose(types, args, kwargs, process_group): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharding_spec/ |
| H A D | api.py | 78 self, tensor: torch.Tensor, src_rank: int = 0, process_group=None argument
113 sharding_spec, op: Callable, types, args, kwargs, process_group argument
180 self, tensor: torch.Tensor, src_rank: int = 0, process_group=None argument
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_common.py | 335 process_group, argument
360 process_group, argument
408 process_group, argument
748 def _test_ddp_with_process_group( argument
829 self, process_group, hook=None, gradient_as_bucket_view=False, state=None argument
846 self, process_group, hook=None, gradient_as_bucket_view=False argument
1121 def _test_sequence_num_incremented(self, process_group, ranks): argument
|
| H A D | test_functional_api.py | 479 def process_group(self): member in TestCollectivesWithNCCL
803 def process_group(self): member in TestFunctionalAutogradWithNCCL
|
| /aosp_15_r20/external/pytorch/torch/distributed/algorithms/model_averaging/ |
| H A D | averagers.py | 24 def __init__(self, process_group=None): argument
88 def __init__(self, period, warmup_steps=0, process_group=None): argument
|
| H A D | hierarchical_model_averager.py | 101 def __init__(self, period_group_size_dict=None, warmup_steps=0, process_group=None): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | comm.cpp | 16 const c10::intrusive_ptr<c10d::ProcessGroup>& process_group, in BroadcastWork()
63 const c10::intrusive_ptr<c10d::ProcessGroup>& process_group, in broadcast_coalesced()
|
| /aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/ |
| H A D | test_sharded_tensor.py | 3033 def my_sharded_asin(types, args, kwargs, process_group): argument
3059 def my_sharded_linear(types, args, kwargs, process_group): argument
3084 def my_op1(types, args, kwargs, process_group, random_param): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/algorithms/ |
| H A D | test_join.py | 89 def __init__(self, device, process_group): argument
159 def process_group(self): member in TestJoin
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_misc.py | 80 def process_group(self): member in TestFSDPMiscMultiProcess
634 def process_group(self): member in TestFSDPMiscMultiThread
|
| /aosp_15_r20/prebuilts/rust/linux-musl-x86/1.81.0/lib/rustlib/src/rust/library/std/src/os/unix/ |
| H A D | process.rs | 196 fn process_group(&mut self, pgroup: i32) -> &mut process::Command; in process_group() method
238 fn process_group(&mut self, pgroup: i32) -> &mut process::Command { in process_group() method
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.80.1/lib/rustlib/src/rust/library/std/src/os/unix/ |
| H A D | process.rs | 196 fn process_group(&mut self, pgroup: i32) -> &mut process::Command; in process_group() method
238 fn process_group(&mut self, pgroup: i32) -> &mut process::Command { in process_group() method
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.81.0.u1/lib/rustlib/src/rust/library/std/src/os/unix/ |
| H A D | process.rs | 196 fn process_group(&mut self, pgroup: i32) -> &mut process::Command; in process_group() method
238 fn process_group(&mut self, pgroup: i32) -> &mut process::Command { in process_group() method
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.81.0/lib/rustlib/src/rust/library/std/src/os/unix/ |
| H A D | process.rs | 196 fn process_group(&mut self, pgroup: i32) -> &mut process::Command; in process_group() method
238 fn process_group(&mut self, pgroup: i32) -> &mut process::Command { in process_group() method
|
| /aosp_15_r20/prebuilts/rust/linux-musl-x86/1.80.1/lib/rustlib/src/rust/library/std/src/os/unix/ |
| H A D | process.rs | 196 fn process_group(&mut self, pgroup: i32) -> &mut process::Command; in process_group() method
238 fn process_group(&mut self, pgroup: i32) -> &mut process::Command { in process_group() method
|
| /aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/trainer/ |
| H A D | ddp_models.py | 4 def basic_ddp_model(self, rank, model, process_group, hook_state, hook): argument
|
| H A D | hook_states.py | 2 def __init__(self, cref, process_group): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fully_shard/ |
| H A D | test_fully_shard_util.py | 36 def process_group(self): member in TestUtils
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | simulate_nccl_errors.py | 31 process_group = c10d.ProcessGroupNCCL(store, rank, world_size) variable
|
| /aosp_15_r20/external/pytorch/test/distributed/checkpoint/ |
| H A D | test_fsdp_model_state.py | 26 def _test_fsdp_model_state(self, process_group) -> None: argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/algorithms/ddp_comm_hooks/ |
| H A D | post_localSGD_hook.py | 37 process_group, argument
|