Home
last modified time | relevance | path

Searched refs:init_method (Results 1 – 25 of 44) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/distributed/rpc/_testing/
H A Dfaulty_agent_backend_registry.py10 init_method, argument
22 init_method=init_method,
/aosp_15_r20/external/pytorch/torch/distributed/rpc/
H A Dbackend_registry.py104 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 Doptions.py85 init_method: str = rpc_contants.DEFAULT_INIT_METHOD,
102 init_method,
H A D__init__.py179 rpc_backend_options.init_method, rank=rank, world_size=world_size
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dddp_under_dist_autograd_test.py333 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 Ddist_utils.py174 def initialize_pg(init_method, rank: int, world_size: int) -> None: argument
179 init_method=init_method,
H A Dcommon_distributed.py484 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 Dsimple_index_file.cc427 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 Dsimple_index_file.h40 SimpleIndex::IndexInitMethod init_method; member
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Dtensorpipe_rpc_agent_test_fixture.py23 init_method=self.init_method,
H A Dfaulty_rpc_agent_test_fixture.py40 init_method=self.init_method,
H A Drpc_test.py643 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 Drpc_agent_test_fixture.py15 def init_method(self): member in RpcAgentTestFixture
/aosp_15_r20/external/pytorch/test/distributed/launcher/bin/
H A Dtest_script_init_method.py50 init_method=args.init_method,
/aosp_15_r20/external/pytorch/torch/_C/
H A D_distributed_rpc.pyi23 init_method: str
27 init_method: str = ...,
80 init_method: str = ...,
H A D_distributed_rpc_testing.pyi12 init_method: str,
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpython_custom_class.cpp25 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__.py30 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 Dfaulty_tensorpipe_agent.h17 std::string init_method,
26 std::move(init_method)), in TensorPipeRpcBackendOptions()
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_store.py195 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 Dtest_c10d_common.py80 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 Dbenchmark_ddp_rpc.py212 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 Dtensorpipe_agent.h85 std::string init_method,
88 : RpcBackendOptions(rpc_timeout, std::move(init_method)), in RpcBackendOptions()
/aosp_15_r20/external/pytorch/docs/source/
H A Ddistributed.rst63 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 Ddistributed_c10d.py1363 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

12