Google Git
Sign in
android / platform / external / pytorch / 848e7ac42a3c121d03bcff6cb53c1afb9b76164e / . / torch / distributed / tensor / parallel
tree: c52742cd2e5a757d7252034c0ba1be91307c3c94 [path history] [tgz]
  1. __init__.py
  2. _data_parallel_utils.py
  3. _utils.py
  4. api.py
  5. ddp.py
  6. fsdp.py
  7. input_reshard.py
  8. loss.py
  9. style.py