Home
last modified time | relevance | path

Searched refs:TensorPipeRpcBackendOptions (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dinit.cpp533 py::class_<TensorPipeRpcBackendOptions>( in rpc_init()
553 &TensorPipeRpcBackendOptions::numWorkerThreads, in rpc_init()
561 &TensorPipeRpcBackendOptions::deviceMaps, in rpc_init()
565 &TensorPipeRpcBackendOptions::devices, in rpc_init()
567 .def("_set_device_map", &TensorPipeRpcBackendOptions::setDeviceMap); in rpc_init()
579 TensorPipeRpcBackendOptions opts, in rpc_init()
H A Dtensorpipe_agent.h79 struct TORCH_API TensorPipeRpcBackendOptions : public RpcBackendOptions { struct
80 TensorPipeRpcBackendOptions(
168 TensorPipeRpcBackendOptions opts,
204 TensorPipeRpcBackendOptions getBackendOptions() const;
328 const TensorPipeRpcBackendOptions opts_;
H A Dtensorpipe_agent.cpp388 TensorPipeRpcBackendOptions opts, in TensorPipeAgent()
1445 TensorPipeRpcBackendOptions TensorPipeAgent::getBackendOptions() const { in getBackendOptions()
/aosp_15_r20/external/pytorch/torch/distributed/rpc/
H A Dbackend_registry.py143 from . import TensorPipeRpcBackendOptions
145 return TensorPipeRpcBackendOptions(
343 from . import TensorPipeAgent, TensorPipeRpcBackendOptions
348 if not isinstance(rpc_backend_options, TensorPipeRpcBackendOptions):
H A Doptions.py46 class TensorPipeRpcBackendOptions(_TensorPipeRpcBackendOptionsBase): class
H A D__init__.py76 from .options import TensorPipeRpcBackendOptions # noqa: F401
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/testing/
H A Dfaulty_tensorpipe_agent.h13 : public TensorPipeRpcBackendOptions {
21 : TensorPipeRpcBackendOptions( in TensorPipeRpcBackendOptions() function
H A Dinit.cpp57 "num_worker_threads", &TensorPipeRpcBackendOptions::numWorkerThreads) in faulty_agent_init()
/aosp_15_r20/external/pytorch/torch/distributed/benchmarks/
H A Dbenchmark_ddp_rpc.py22 from torch.distributed.rpc import RRef, TensorPipeRpcBackendOptions
211 rpc_backend_options = TensorPipeRpcBackendOptions()
/aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/
H A Dlauncher.py24 from torch.distributed.rpc import TensorPipeRpcBackendOptions
298 rpc_backend_options = TensorPipeRpcBackendOptions(rpc_timeout=args.rpc_timeout)
/aosp_15_r20/external/pytorch/test/cpp/rpc/
H A Dtest_e2e_tensorpipe.cpp24 TensorPipeRpcBackendOptions opts( in buildRpcAgent()
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/
H A D__init__.py39 rpc_backend_options = rpc.TensorPipeRpcBackendOptions(_transports=tp_transports())
/aosp_15_r20/external/pytorch/docs/source/
H A Drpc.rst181 >>> rpc_backend_options=rpc.TensorPipeRpcBackendOptions(
189 .. autoclass:: TensorPipeRpcBackendOptions
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_store.py194 rpc_backend_options = rpc.TensorPipeRpcBackendOptions()
359 backend_opts = rpc.TensorPipeRpcBackendOptions(
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Drpc_test.py4652 rpc_backend_options = rpc.TensorPipeRpcBackendOptions(
4671 rpc_backend_options = rpc.TensorPipeRpcBackendOptions(
4694 rpc_backend_options = rpc.TensorPipeRpcBackendOptions(
4720 rpc_backend_options = rpc.TensorPipeRpcBackendOptions(
5138 rpc_backend_options = rpc.TensorPipeRpcBackendOptions(init_method="env://")
5147 … rpc_backend_options = rpc.TensorPipeRpcBackendOptions(init_method="tcp://127.0.0.1:23456")
5638 rpc_backend_options=rpc.TensorPipeRpcBackendOptions(
6268 options = rpc.TensorPipeRpcBackendOptions(
/aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/
H A Dtest_sharded_tensor.py1101 rpc_backend_options = rpc.TensorPipeRpcBackendOptions(