Searched refs:rank_by_s_d (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce.py | 242 rank_by_s_d = [[-1 for d in range(0, devices)]
248 rank_by_s_d[s][d] = t
251 return (pred_by_s_d, rank_by_s_d)
281 (pred_by_s_d, rank_by_s_d) = _ring_permutations(
285 num_subchunks, pred_by_s_d, rank_by_s_d, red_op)
288 output_tensors = _build_ring_scatter(pred_by_s_d, rank_by_s_d,
298 pred_by_s_d, rank_by_s_d, red_op): argument
342 rank = rank_by_s_d[s][d]
352 rank = rank_by_s_d[s][d]
377 def _build_ring_scatter(pred_by_s_d, rank_by_s_d, argument
[all …]
|