Google Git
Sign in
android / platform / external / pytorch / 35aeb6aa8533cf9bb88c0f8375b9ba8be20b2425 / . / test / distributed / tensor / parallel
tree: be25907f0b7ee73a82906436a15eff4b82ed6cd2 [path history] [tgz]
  1. __init__.py
  2. test_ddp_2d_parallel.py
  3. test_fsdp_2d_parallel.py
  4. test_parallelize_api.py
  5. test_tp_examples.py
  6. test_tp_random_state.py
  7. test_tp_style.py
  8. test_view_sharding_dim_change.py