Home
last modified time | relevance | path

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__.py89 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 Dbackend_registry.py341 store, name, rank, world_size, rpc_backend_options argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Dtensorpipe_rpc_agent_test_fixture.py20 def rpc_backend_options(self): member in TensorPipeRpcAgentTestFixture
H A Drpc_agent_test_fixture.py37 def rpc_backend_options(self): member in RpcAgentTestFixture
H A Dfaulty_rpc_agent_test_fixture.py37 def rpc_backend_options(self): member in FaultyRpcAgentTestFixture
H A Drpc_test.py124 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 Dfaulty_agent_backend_registry.py30 store, name, rank, world_size, rpc_backend_options argument
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/
H A Dlauncher.py238 def run_master(rank, data, args, extra_configs, rpc_backend_options): argument