Searched defs:rpc_backend_options (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/pytorch/torch/distributed/rpc/ |
H A D | __init__.py | 89 rpc_backend_options=None, argument 203 def _validate_rpc_args(backend, store, name, rank, world_size, rpc_backend_options): argument 225 rpc_backend_options=None, argument
|
H A D | backend_registry.py | 341 store, name, rank, world_size, rpc_backend_options argument
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
H A D | tensorpipe_rpc_agent_test_fixture.py | 20 def rpc_backend_options(self): member in TensorPipeRpcAgentTestFixture
|
H A D | rpc_agent_test_fixture.py | 37 def rpc_backend_options(self): member in RpcAgentTestFixture
|
H A D | faulty_rpc_agent_test_fixture.py | 37 def rpc_backend_options(self): member in FaultyRpcAgentTestFixture
|
H A D | rpc_test.py | 124 def _stub_init_rpc_backend_handler(store, name, rank, world_size, rpc_backend_options): argument
|
/aosp_15_r20/external/pytorch/torch/distributed/rpc/_testing/ |
H A D | faulty_agent_backend_registry.py | 30 store, name, rank, world_size, rpc_backend_options argument
|
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/ |
H A D | launcher.py | 238 def run_master(rank, data, args, extra_configs, rpc_backend_options): argument
|