Searched refs:RemoteModule (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/nn/api/ |
| H A D | remote_module_test.py | 11 from torch.distributed.nn import RemoteModule
114 remote_module = RemoteModule(remote_device, MyModule, args, kwargs)
143 RemoteModule(remote_device, BadModule, args, kwargs).forward()
149 RemoteModule(remote_device, BadModule, args, kwargs).forward()
555 RemoteModule.init_from_module_rref,
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/api/ |
| H A D | remote_module.py | 581 remote_module = object.__new__(RemoteModule)
605 class RemoteModule(_RemoteModule): class
698 m = object.__new__(RemoteModule)
759 _internal_rpc_pickler._register_reducer(RemoteModule, _remote_module_reducer)
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | rpc.rst | 236 RemoteModule chapter
240 RemoteModule is not currently supported when using CUDA tensors
242 ``RemoteModule`` is an easy way to create an nn.Module remotely on a different
246 asynchronously if needed via additional APIs supported by RemoteModule.
248 .. autoclass:: torch.distributed.nn.api.remote_module.RemoteModule
297 …<https://pytorch.org/tutorials/advanced/rpc_ddp_tutorial.html>`__ (covers **RemoteModule** as well)
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | ddp_under_dist_autograd_test.py | 15 from torch.distributed.nn import RemoteModule
630 remote_layer1 = RemoteModule(
679 remote_layer1 = RemoteModule(
689 remote_layer3 = RemoteModule(
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/ |
| H A D | __init__.py | 7 from .api.remote_module import RemoteModule
|