Google Git
Sign in
android / platform / external / pytorch / bc0c74bcd5 / . / torch / distributed / tensor / parallel
tree: c841fa926f5a2293eeec8c21dd97c296f2f6a87a [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