Google Git
Sign in
android / platform / external / pytorch / e8d3c4be36 / . / torch / distributed / _shard
tree: 299d083bc3d71ac2482892322138b5b8df7ab2f3 [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