Google Git Google Git
Sign in
android/platform/external/pytorch/cf12edee02/./test/distributed/tensor/parallel
tree: 4280f889640356c48e6e046d19cbaae7a329a3ae [search]
  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