Searched refs:red_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce.py | 255 gpu_perm, red_op, un_op=None): argument
285 num_subchunks, pred_by_s_d, rank_by_s_d, red_op)
298 pred_by_s_d, rank_by_s_d, red_op): argument
346 new_partial_reductions[chunk_index] = red_op(
426 def build_recursive_hd_all_reduce(input_tensors, red_op, un_op=None): argument
465 reduced_shards = _build_recursive_hd_gather(input_tensors, devices, red_op)
474 def _build_recursive_hd_gather(input_tensors, devices, red_op): argument
508 new_chunks[d] = red_op(left_split[0], right_split[0])
510 new_chunks[d + span] = red_op(left_split[1], right_split[1])
552 def build_shuffle_all_reduce(input_tensors, gather_devices, red_op, un_op=None): argument
[all …]
|