Google Git
Sign in
android / platform / external / pytorch / 007e12a3e9 / . / torch / distributed / _shard
tree: 2a692fb57703fd2dbc40d53523d42eaa8b15317d [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