Searched refs:upper_level_f (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce.py | 708 def _build_nccl_hybrid(input_tensors, red_op, upper_level_f): argument
743 level_2_output = upper_level_f(up_values)
777 def upper_level_f(x): function
779 return _build_nccl_hybrid(input_tensors, red_op, upper_level_f)
784 upper_level_f = lambda x: build_recursive_hd_all_reduce(x, red_op, un_op) function
785 return _build_nccl_hybrid(input_tensors, red_op, upper_level_f)
791 def upper_level_f(x): function
794 return _build_nccl_hybrid(input_tensors, nccl_red_op, upper_level_f)
797 def _build_shuffle_hybrid(input_tensors, gather_devices, red_op, upper_level_f): argument
828 level_2_output = upper_level_f(up_values)
[all …]
|