Google Git
Sign in
android / platform / external / pytorch / e1e056ac44 / . / torch / distributed / fsdp
tree: eaf862ff302357f32a2b8594776ec3d42538703f [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