Google Git
Sign in
android / platform / external / pytorch / db8abde9b6c4735d18d4681a1f70a55ff0b09f5b / . / torch / distributed / fsdp
tree: 2710f45aa07816b5112cfc09fad75cd63b59ad19 [path history] [tgz]
  1. __init__.py
  2. _common_utils.py
  3. _dynamo_utils.py
  4. _exec_order_utils.py
  5. _fsdp_extensions.py
  6. _init_utils.py
  7. _limiter_utils.py
  8. _optim_utils.py
  9. _runtime_utils.py
  10. _shard_utils.py
  11. _state_dict_utils.py
  12. _trace_utils.py
  13. _traversal_utils.py
  14. _unshard_param_utils.py
  15. _utils.py
  16. _wrap_utils.py
  17. api.py
  18. flat_param.py
  19. fully_sharded_data_parallel.py
  20. sharded_grad_scaler.py
  21. wrap.py