pytorch/torch/distributed/tensor/parallel
2023-02-14 04:52:04 +00:00
..
__init__.py [PT-D][Tensor parallelism] Add documentations for TP (#94421) 2023-02-09 02:31:06 +00:00
_utils.py [BE] Merge isinstance calls together (#94419) 2023-02-09 00:47:26 +00:00
_view_with_dim_change.py [tp] umft distributed.tensor.parallel (#89969) 2022-12-01 20:58:16 +00:00
api.py [tp] additional doc fixes (#94786) 2023-02-14 04:52:04 +00:00
fsdp.py [tp] umft distributed.tensor.parallel (#89969) 2022-12-01 20:58:16 +00:00
multihead_attention_tp.py [BE] [2/3] Rewrite super() calls in functorch and torch (#94588) 2023-02-10 21:16:33 +00:00
style.py [PT-D][Tensor parallelism] Add documentations for TP (#94421) 2023-02-09 02:31:06 +00:00