Google Git
Sign in
android / platform / external / pytorch / 03b9569d2c / . / torch / distributed / checkpoint
tree: 2aa62517e0a79d4260b33c9b82c9a4f97dbb4e0f [path history] [tgz]
  1. __init__.py
  2. _dedup_tensors.py
  3. _nested_dict.py
  4. _sharded_tensor_utils.py
  5. _traverse.py
  6. api.py
  7. default_planner.py
  8. filesystem.py
  9. metadata.py
  10. optimizer.py
  11. planner.py
  12. planner_helpers.py
  13. resharding.py
  14. state_dict_loader.py
  15. state_dict_saver.py
  16. storage.py
  17. utils.py