Google Git
Sign in
android / platform / external / pytorch / 0a651a231d81fc34d6b8858ee41cf7b9772c042a / . / torch / distributed / _shard
tree: 828fd1d8c19ec8d854f89238308e4748fcac198d [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