Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py2633 def _test_all_reduce_helper( member in DistributedTest._DistTestBase
2694 self._test_all_reduce_helper(
2709 self._test_all_reduce_helper(
2729 self._test_all_reduce_helper(
2750 self._test_all_reduce_helper(
2768 self._test_all_reduce_helper(
2809 self._test_all_reduce_helper(
2827 self._test_all_reduce_helper(
2842 self._test_all_reduce_helper(
2851 self._test_all_reduce_helper(
[all …]