Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_gloo.py47 simple_sparse_reduce_tests,
645 tests = simple_sparse_reduce_tests(
672 tests = simple_sparse_reduce_tests(self.rank, self.world_size, num_inputs=1)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_distributed.py416 def simple_sparse_reduce_tests(rank: int, world_size: int, num_inputs: int = 1): function
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py64 simple_sparse_reduce_tests,
2959 tests = simple_sparse_reduce_tests(