Google Git
Sign in
android / platform / external / pytorch / bc0c74bcd5 / . / torch / distributed / _shard
tree: d80450fbf9e31f604ff887a87f14be0f18566723 [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. sharder.py