pytorch
f612d4eb - Add 'remote_parameters' and 'get_module_rref' to RemoteModule docs. (#54645)

Commit
4 years ago
Add 'remote_parameters' and 'get_module_rref' to RemoteModule docs. (#54645) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54645 Had to replace RRef[..] with just RRef in the return signature since sphynx seemed to completely mess up rendering RRef[..] ghstack-source-id: 125024783 Test Plan: View locally. Reviewed By: SciPioneer Differential Revision: D27314609 fbshipit-source-id: 2dd9901e79f31578ac7733f79dbeb376f686ed75
Author
Parents
  • docs/source
    • File
      rpc.rst
  • torch/distributed/nn/api
    • File
      remote_module.py