Google Git
Sign in
android / platform / external / pytorch / db8abde9b6c4735d18d4681a1f70a55ff0b09f5b / . / torch / distributed / tensor / parallel
tree: 1e62a4932d3f0dfda969752ea2835b30c0b3626c [path history] [tgz]
  1. __init__.py
  2. _utils.py
  3. _view_with_dim_change.py
  4. api.py
  5. fsdp.py
  6. multihead_attention_tp.py
  7. style.py