Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_snode_runtime.py295 def test_reduce_scatter_tensor(self): member in TestCommAnalysis
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_functional_api.py322 def test_reduce_scatter_tensor(self, device): member in TestTraceableCollectives
755 def test_reduce_scatter_tensor(self, compile: bool) -> None: member in TestFunctionalAutograd
H A Dtest_c10d_gloo.py702 def test_reduce_scatter_tensor(self): member in ProcessGroupGlooTest