Google Git
Sign in
android / platform / external / pytorch / 5c16dfd708 / . / test / distributed / tensor / parallel
tree: 24b53081c963d85f24ee786e68dcc1c3ca206307 [path history] [tgz]
  1. __init__.py
  2. test_2d_parallel.py
  3. test_parallelize_api.py
  4. test_tp_examples.py
  5. test_tp_style.py
  6. test_view_sharding_dim_change.py