Searched refs:_test_sparse_all_reduce_sum (Results 1 – 1 of 1) sorted by relevance
2956 def _test_sparse_all_reduce_sum(self, fn): member in DistributedTest._DistTestBase2971 self._test_sparse_all_reduce_sum(lambda t: t)2978 self._test_sparse_all_reduce_sum(lambda t: t.clone().cuda())