_patch_two_tensor_fsdp_all_gather (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A D	test_fully_shard_extensions.py	`69 def _patch_two_tensor_fsdp_all_gather(self): member in TestFullyShardAllGatherExtensionsCommon 103 with self._patch_two_tensor_fsdp_all_gather(): 151 with self._patch_two_tensor_fsdp_all_gather():`