Home
last modified time | relevance | path

Searched +path:parallel +path:distributed (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A Ddistributed.py
/aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/
H A Dddp.py
H A Dstyle.py
H A D_data_parallel_utils.py
H A Dfsdp.py
H A Dloss.py
H A Dinput_reshard.py
H A Dapi.py
H A D__init__.py
H A D_utils.py
/aosp_15_r20/external/pytorch/test/distributed/tensor/parallel/
H A Dtest_tp_examples.py
H A Dtest_micro_pipeline_tp.py
H A D__init__.py
H A Dtest_parallelize_api.py
H A Dtest_tp_style.py
H A Dtest_tp_random_state.py
/aosp_15_r20/external/pytorch/docs/source/
H A Ddistributed.tensor.parallel.rst