_get_ref_num_reduce_scatters (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A D	test_fsdp_comm.py	`91 def _get_ref_num_reduce_scatters( member in TestCommunication 266 ref_num_reduce_scatters = self._get_ref_num_reduce_scatters( 284 ref_num_reduce_scatters = self._get_ref_num_reduce_scatters(`