Google Git
Sign in
android / platform / external / pytorch / f21e77d9a6 / . / torch / distributed / fsdp
tree: ae9c1e9fc33d0304b948881c0ac03f9fe0dea209 [path history] [tgz]
  1. __init__.py
  2. _optim_utils.py
  3. _shard_utils.py
  4. _symbolic_trace.py
  5. _utils.py
  6. flat_param.py
  7. flatten_params_wrapper.py
  8. fully_sharded_data_parallel.py
  9. sharded_grad_scaler.py
  10. utils.py
  11. wrap.py