Google Git
Sign in
android / platform / external / pytorch / a360be50b5 / . / torch / distributed / _shard
tree: 5489631265168ca1cbf12d7fde068812ec5c77c3 [path history] [tgz]
  1. checkpoint/
  2. sharded_optim/
  3. sharded_tensor/
  4. sharding_plan/
  5. sharding_spec/
  6. __init__.py
  7. _utils.py
  8. api.py
  9. common_op_utils.py
  10. metadata.py
  11. op_registry_utils.py
  12. partial_tensor.py
  13. replicated_tensor.py
  14. sharder.py