Searched refs:_ring_permutations (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce_test.py | 49 pred_by_c_d, rank_by_c_d = ar._ring_permutations(1, 0, [])
53 pred_by_c_d, rank_by_c_d = ar._ring_permutations(1, 1, [0])
56 pred_by_c_d, rank_by_c_d = ar._ring_permutations(1, 1, [0, 1, 2])
60 pred_by_c_d, rank_by_c_d = ar._ring_permutations(2, 1, [0, 1, 2])
63 pred_by_c_d, rank_by_c_d = ar._ring_permutations(3, 1, [0, 1, 2])
66 pred_by_c_d, rank_by_c_d = ar._ring_permutations(2, 1, [2, 1, 0])
70 pred_by_c_d, rank_by_c_d = ar._ring_permutations(1, 2, [0, 1, 2, 3])
73 pred_by_c_d, rank_by_c_d = ar._ring_permutations(1, 4, [0, 1, 2, 3])
79 pred_by_c_d, rank_by_c_d = ar._ring_permutations(2, 2, [0, 1, 2, 3])
84 pred_by_c_d, rank_by_c_d = ar._ring_permutations(2, 2, [0, 3, 2, 1])
[all …]
|
| D | all_reduce.py | 193 def _ring_permutations(num_workers, num_subchunks, gpu_perm): function
281 (pred_by_s_d, rank_by_s_d) = _ring_permutations(
|