Searched refs:_ring_permutations (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | all_reduce_test.py | 48 pred_by_c_d, rank_by_c_d = ar._ring_permutations(1, 0, [])
52 pred_by_c_d, rank_by_c_d = ar._ring_permutations(1, 1, [0])
55 pred_by_c_d, rank_by_c_d = ar._ring_permutations(1, 1, [0, 1, 2])
59 pred_by_c_d, rank_by_c_d = ar._ring_permutations(2, 1, [0, 1, 2])
62 pred_by_c_d, rank_by_c_d = ar._ring_permutations(3, 1, [0, 1, 2])
65 pred_by_c_d, rank_by_c_d = ar._ring_permutations(2, 1, [2, 1, 0])
69 pred_by_c_d, rank_by_c_d = ar._ring_permutations(1, 2, [0, 1, 2, 3])
72 pred_by_c_d, rank_by_c_d = ar._ring_permutations(1, 4, [0, 1, 2, 3])
78 pred_by_c_d, rank_by_c_d = ar._ring_permutations(2, 2, [0, 1, 2, 3])
83 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(
|