Searched refs:pred_by_s_d (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce.py | 240 pred_by_s_d = [[-1 for d in range(0, devices)]
249 pred_by_s_d[s][d] = perms_by_s[s][(t + devices - 1) % devices]
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
344 pred_dev = pred_by_s_d[s][d]
377 def _build_ring_scatter(pred_by_s_d, rank_by_s_d, argument
408 pred_dev = pred_by_s_d[s][d]
|