/aosp_15_r20/external/pytorch/torch/distributed/rpc/_testing/ |
H A D | faulty_agent_backend_registry.py | 10 init_method, argument 22 init_method=init_method,
|
/aosp_15_r20/external/pytorch/torch/distributed/rpc/ |
H A D | backend_registry.py | 104 init_method=rpc_constants.DEFAULT_INIT_METHOD, argument 108 rpc_timeout, init_method, **kwargs 137 init_method, argument 147 init_method=init_method,
|
H A D | options.py | 85 init_method: str = rpc_contants.DEFAULT_INIT_METHOD, 102 init_method,
|
H A D | __init__.py | 179 rpc_backend_options.init_method, rank=rank, world_size=world_size
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
H A D | ddp_under_dist_autograd_test.py | 333 init_method=INIT_METHOD_TEMPLATE.format(file_name=self.file_name), 355 init_method=INIT_METHOD_TEMPLATE.format(file_name=self.file_name), 372 init_method=INIT_METHOD_TEMPLATE.format(file_name=self.file_name), 522 init_method=INIT_METHOD_TEMPLATE.format(file_name=f"{self.file_name}_pg"), 589 init_method=INIT_METHOD_TEMPLATE.format(file_name=self.file_name), 622 init_method=INIT_METHOD_TEMPLATE.format(file_name=self.file_name), 674 init_method=INIT_METHOD_TEMPLATE.format(file_name=self.file_name),
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | dist_utils.py | 174 def initialize_pg(init_method, rank: int, world_size: int) -> None: argument 179 init_method=init_method,
|
H A D | common_distributed.py | 484 def initialize_temp_directories(init_method: Optional[str] = None) -> None: 493 if init_method is not None: 494 os.environ["INIT_METHOD"] = init_method
|
/aosp_15_r20/external/cronet/net/disk_cache/simple/ |
H A D | simple_index_file.cc | 427 out_result->init_method = SimpleIndex::INITIALIZE_METHOD_LOADED; in SyncLoadIndexEntries() 428 UmaRecordIndexInitMethod(out_result->init_method, cache_type); in SyncLoadIndexEntries() 443 out_result->init_method = SimpleIndex::INITIALIZE_METHOD_RECOVERED; in SyncLoadIndexEntries() 458 out_result->init_method = SimpleIndex::INITIALIZE_METHOD_NEWCACHE; in SyncLoadIndexEntries() 463 UmaRecordIndexInitMethod(out_result->init_method, cache_type); in SyncLoadIndexEntries()
|
H A D | simple_index_file.h | 40 SimpleIndex::IndexInitMethod init_method; member
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
H A D | tensorpipe_rpc_agent_test_fixture.py | 23 init_method=self.init_method,
|
H A D | faulty_rpc_agent_test_fixture.py | 40 init_method=self.init_method,
|
H A D | rpc_test.py | 643 def __init__(self, init_method): argument 647 self.init_method = init_method 1385 self.init_method, rank=self.rank, world_size=self.world_size 1444 init_method=self.file_init_method, 3373 init_method=self.file_init_method, 4321 init_method=self.init_method, 4360 init_method=self.init_method, 4419 new_backend_options.init_method += "init_2" 4458 rpc_backend_options={"init_method": self.init_method} 4465 rpc_backend_options = FooBackendOptions(self.init_method) [all …]
|
H A D | rpc_agent_test_fixture.py | 15 def init_method(self): member in RpcAgentTestFixture
|
/aosp_15_r20/external/pytorch/test/distributed/launcher/bin/ |
H A D | test_script_init_method.py | 50 init_method=args.init_method,
|
/aosp_15_r20/external/pytorch/torch/_C/ |
H A D | _distributed_rpc.pyi | 23 init_method: str 27 init_method: str = ..., 80 init_method: str = ...,
|
H A D | _distributed_rpc_testing.pyi | 12 init_method: str,
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
H A D | python_custom_class.cpp | 25 Method init_method(instance._ivalue(), init_fn); in __call__() local 26 invokeScriptMethodFromPython(init_method, args, kwargs); in __call__()
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/ |
H A D | __init__.py | 30 init_method=f"file://{self.file_name}", 40 rpc_backend_options.init_method = f"file://{self.file_name}"
|
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/testing/ |
H A D | faulty_tensorpipe_agent.h | 17 std::string init_method, 26 std::move(init_method)), in TensorPipeRpcBackendOptions()
|
/aosp_15_r20/external/pytorch/test/distributed/ |
H A D | test_store.py | 195 rpc_backend_options.init_method = f"file://{file.name}" 203 "gloo", rank=0, world_size=1, init_method=f"file://{file.name}" 354 init_method="env://", 360 init_method=f"tcp://{addr}:{port}", _transports=tp_transports() 1017 init_method=f"tcp://{DEFAULT_HOSTNAME}:{port}?use_libuv=0", 1026 dist.init_process_group("gloo", rank=0, world_size=1, init_method="env://")
|
H A D | test_c10d_common.py | 80 def _test_store_timeout(self, backend, init_method, c2p): argument 84 init_method=init_method, 112 for init_method in self._init_methods(): 115 target=self._test_store_timeout, args=(backend, init_method, c2p)
|
/aosp_15_r20/external/pytorch/torch/distributed/benchmarks/ |
H A D | benchmark_ddp_rpc.py | 212 rpc_backend_options.init_method = "tcp://localhost:29500" 266 init_method="tcp://localhost:29501",
|
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ |
H A D | tensorpipe_agent.h | 85 std::string init_method, 88 : RpcBackendOptions(rpc_timeout, std::move(init_method)), in RpcBackendOptions()
|
/aosp_15_r20/external/pytorch/docs/source/ |
H A D | distributed.rst | 63 If the `init_method` argument of :func:`init_process_group` points to a file it must adhere 66 - Local file system, ``init_method="file:///d:/tmp/some_file"`` 67 - Shared file system, ``init_method="file://////{machine_name}/{share_folder_name}/some_file"`` 219 dist.init_process_group(backend, init_method='tcp://10.1.1.20:23456', 261 dist.init_process_group(backend, init_method='file:///mnt/nfs/sharedfile', 278 This is the default method, meaning that ``init_method`` does not have to be specified (or 335 as an alternative to specifying ``init_method``.) There are 3 choices for
|
/aosp_15_r20/external/pytorch/torch/distributed/ |
H A D | distributed_c10d.py | 1363 init_method: Optional[str] = None, 1470 init_method is None 1476 elif init_method is None: 1477 init_method = "env://" 1518 not_none(init_method), rank, world_size, timeout=timeout 1543 _default_pg_init_method = init_method
|