Home
last modified time | relevance | path

Searched refs:test_tp_with_fsdp_offloading (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/_composable/test_composability/
H A Dtest_2d_composability.py176 def test_tp_with_fsdp_offloading(self): member in TestFullyShard2DTraining
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_training.py939 def test_tp_with_fsdp_offloading(self): member in TestFullyShard2DTraining