Searched defs:red_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | all_reduce.py | 255 gpu_perm, red_op, un_op=None): argument
298 pred_by_s_d, rank_by_s_d, red_op): argument
426 def build_recursive_hd_all_reduce(input_tensors, red_op, un_op=None): argument
480 def _build_recursive_hd_gather(input_tensors, devices, red_op): argument
558 def build_shuffle_all_reduce(input_tensors, gather_devices, red_op, un_op=None): argument
592 def _build_shuffle_gather(input_tensors, gather_devices, red_op, un_op=None): argument
685 def build_nccl_all_reduce(input_tensors, red_op, un_op=None): argument
714 def _build_nccl_hybrid(input_tensors, red_op, upper_level_f): argument
779 def build_nccl_then_ring(input_tensors, subdiv, red_op, un_op=None): argument
788 def build_nccl_then_recursive_hd(input_tensors, red_op, un_op=None): argument
[all …]
|